可以看到,当tensor全是double型时,int函数会把所有元素取整,从1.5可以看出,不是四舍五入,而是取整。double函数又把整数型元素变成double型。
th> a
0.0000 2.0000 0.0000
0.0000 1.5000 0.0000
[torch.DoubleTensor of size 2x3] [.0002s]
th> a:int() [torch.IntTensor of size 2x3] [.0002s]
th> a:double()
0.0000 2.0000 0.0000
0.0000 1.5000 0.0000
[torch.DoubleTensor of size 2x3]