fix np.linspace core dump

This commit is contained in:
yanglf1121 2020-12-18 18:43:12 +08:00
parent 760cd1870a
commit dc2f33d0d1
1 changed files with 2 additions and 2 deletions

View File

@ -454,10 +454,10 @@ def linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None, axis
if retstep:
array_out, step_out = out[0], out[1]
tensor_out = Tensor.from_numpy(array_out)
tensor_out = Tensor(array_out)
return tensor_out, step_out
tensor_out = Tensor.from_numpy(out)
tensor_out = Tensor(out)
return tensor_out