一、张量的概念
张量是一个多维数组,它是标量、向量、矩阵的高维拓展。
0维张量,没有方向;1维张量,数据沿着一个方向排列存放;矩阵是2维张量数据,有行和列两个维度,在图像中通常表示灰度图片;3维张量,通常表示彩色图片,有长、宽、颜色通道三个维度。
1)Tensor与Variable
Variable是torch.autograd中的数据类型,主要用于封装Tensor,进行自动求导。
torch.autograd.Variable()
data:被包装的Tensor
grad:data的梯度
grad_fn:创建Tensor的Function,是自动求导的关键
requires_grad:指示是否需要梯度
is_leaf:指示是否是叶子节点(张量)
pytorch0.4.0版开始,Variable并入Tensor。
**torch.Tensor()**除了上面五个属性,还加入了新的3个属性。
dtype:张量的数据类型,如:torch.FloatTensor;torch.cuda.FloatTensor。
shape:张量的形状,如(64,3,224,224)
device:张量所在设备,GPU/CPU
学习pytorch第一天——张量简介与创建
最新推荐文章于 2023-02-18 16:59:02 发布