pytorch 生成全零tensor 指定type_1.1 PyTorch中的张量

1 张量的概念

张量(Tensor)是一个多维数组,它是标量、向量、矩阵的高维拓展。

1.1 张量与Variable的渊源

Variable是torch.autograd中的数据类型。

Variable主要用于封装Tensor,进行自动求导。

data: 被包装的张量
grad: data的梯度
grad_fn: 记录创建该张量时所用的方法(函数)
require_grad: 指示是否需要梯度
is_leaf: 指示是否是叶子结点(张量),用于计算图

从PyTorch0.4.0开始,Variable并入张量。张量是Variable的5个属性再加上以下与数据相关的3个属性。

dtype: 张量的数据类型,如torch.FloatTensor,torch.cuda.FloatTensor
shape: 张量的形状,如四阶张量(64,3,224,224)
device: 张量所在设备,GPU/CPU,是加速的关键

v2-558a4cfe792e73e68d0e6f917008519e_b.jpg

重要的数据类型是:torch.float32(卷积层权值),torch.long(图像标签)

PS: 什么是四阶张量?

以图像为例,如下结构为四阶张量。

  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值