double函数和int函数

 

可以看到，当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]

                                                                      [0.0002s]
th> a:int()
 0  2  0
 0  1  0
[torch.IntTensor of size 2x3]

                                                                      [0.0002s]
th> a:double()
 0.0000  2.0000  0.0000
 0.0000  1.5000  0.0000
[torch.DoubleTensor of size 2x3]