1. Tensor

张量是什么?
  1. 张量:多维数组。

  2. 张量的属性:
    dtype:张量的数据类型
    shape:张量的形状

    data:被包装的tensor
    grad:data的梯度
    grad_fn:创建Tensor的Function
    requires_grad:指示是否需要梯度
    is_leaf:指示是否是叶子节点
    device:张量所在设备

张量的创建
torch.tensor(data, dtype=None, device=None, requires_grad=False)
torch.zeros(size)		
torch.zeros_like(input)   # input需要是一个tensor,输出一个相同shape的tensor

torch.ones()
torch.ones_like()
torch.arange(start=0, end, step=1)    # 等差的一维张量
torch.linspace(start=0, end, step=1)  # 均分的一维张量
torch.normal(mean, std)                   # 正态分布
 # 在区间[0,1)上,生成均匀分布
torch.rand()                              
torch.rand_like()
# 在[low,high)上生成整数的均匀分布
torch,randint()
torch.randint_like()
# 张量拼接
torch.cat((a,b), dim=0, out=None)
# 压缩长度为1的维度
torch.squeeze(imput, dim=None, out=None)
# 依据dim扩展维度,长度为1
torch.unsqueeze(input, dim, out=None)
# 张量对应元素相加/减/乘/除
torch.add(a,b)
torch.sub(a,b)
torch.mul(a,b)
torch.div(a,b)
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值