Tensor开方报错

。

RuntimeError: "sqrt" not implemented for 'torch.LongTensor' 

tensor1 = t.tensor([[1,2,3],
              [4,5,6]])
print(tensor1)
print(tensor1.sqrt())
print(t.tensor([1,2,3,
              4,5,6])**0.5)
print(t.tensor([1,2,3,
              4,5,6],dtype = t.float64)**0.5)

因为默认生成的是LongTensor，得出的开方结果都是[1,1,1,1,1,1]。

所以设置成float64可以得出正确的结果。

另外，LongTensor直接调用sqrt()函数会报错，LongTensor没有这个函数。