2020
10-10
10-10
PyTorch中Tensor的数据类型和运算的使用
在使用Tensor时,我们首先要掌握如何使用Tensor来定义不同数据类型的变量。Tensor时张量的英文,表示多维矩阵,和numpy对应,PyTorch中的Tensor可以和numpy的ndarray相互转换,唯一不同的是PyTorch可以在GPU上运行,而numpy的ndarray只能在cpu上运行。 常用的不同数据类型的Tensor,有32位的浮点型torch.FloatTensor, 64位浮点型torch.DoubleTensor, 16位整形...
继续阅读 >