How to use the redisai.Tensor function in redisai

To help you get started, we’ve selected a few redisai examples, based on popular ways it is used in public projects.

Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.

github RedisAI / redisai-py / test / test_model.py View on Github external
def testONNXGraph(self):
        onnx_model = get_onnx_model()
        path = f'{time.time()}.onnx'
        save_model(onnx_model, path)
        model = load_model(path)
        os.remove(path)
        con = self.get_client()
        con.modelset('onnxmodel', Backend.onnx, Device.cpu, model)
        con.tensorset('a', Tensor.scalar(DType.float, 2, -1))
        con.modelrun('onnxmodel', ['a'], ['c'])
        tensor = con.tensorget('c')
        self.assertEqual([2.0, 0.0], tensor.value)
github RedisAI / redisai-examples / python_client / torch_chatbot / redis_db.py View on Github external
np.array(utils.SOS_token, dtype=np.int64).reshape(1, 1))
        self.con.tensorset('d_input', sos_tensor)
        i = 0
        out = []
        while i < self.max_len:
            i += 1
            self.con.modelrun(
                'decoder',
                input=['d_input', 'd_hidden', 'e_output'],
                output=['d_output', 'd_hidden'])
            d_output = self.con.tensorget('d_output', as_type=rai.BlobTensor).to_numpy()
            d_output_ret = d_output.reshape(1, utils.voc.num_words)
            ind = int(d_output_ret.argmax())
            if ind == utils.EOS_token:
                break
            inter_tensor = rai.Tensor(rai.DType.int64, shape=[1, 1], value=ind)
            self.con.tensorset('d_input', inter_tensor)
            if ind == utils.PAD_token:
                continue
            out.append(ind)
        return utils.indices2str(out)