Tensor 张量
- 张量类似于多维数组,它是标量、向量、矩阵的高维拓展;
- 可以将标量看做0维张量,数组看做是一维张量,矩阵看做是一个二维张量,以此类比理解。
Tensor与Variable
- Variable是Pytorch0.4.0之前的数据类型,在该版本之后已经并入到Tensor中。
- Variable是Torch.autograd中的数据类型,主要封装于Tensor,进行自动求导。
-
Variable中有五个属性:data、grad、grad_fn、requires_grad、is_leaf。
Tensor
- Pytorch0.4.0版开始,Variable并入Tensor
- 除过Variable的数据类型,还有以下:
- dtype:张量的数据类型,如torch.FloatTensor,torch.cuda.FloatTensor
- shape:张量的形状。
-
device:张量所在的设备,GPU/CPU,是加速的关键。
网友评论