版权声明:欢迎大佬指正! https://blog.csdn.net/sinat_36215255/article/details/86310531
1.Terson 博客
Tensor类似于numpy的ndarrays,另外它还可以在GPU上使用Tensor进行加速计算,其实就是一个多维数组,我们可以对其进行各种数学运算和处理
import torch
x = torch.Tensor(5, 3)
print(x)
#输出:
0.0000 0.0000 0.0000
0.0000 0.4677 0.0000
0.4692 0.0000 0.4662
0.0000 0.4693 0.0000
12.7686 189.2708 0.0000
[torch.FloatTensor of size 5x3]
terson与numpy是可以相互转化的。
2.Variable博客2
tensor是PyTorch中的完美组件,但是构建神经网络还远远不够,我们需要能够构建计算图的tensor,这就是Variable。Variable是对tensor的封装,操作和tensor是一样的,但是每个Variable都有三个属性,Variable中的tensor本身.data,对应tensor的梯度.grad以及这个Variable是通过说明方式得到的.grad_fn