Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
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)
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)