首页 > 代码库 > double函数和int函数

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]

 

double函数和int函数