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沒有這個函數。