Pytorch | 详解Pytorch科学计算包——Tensor

本文始发于个人公众号:TechFlow,原创不易,求个关注


今天是Pytorch专题的第二篇,我们继续来了解一下Pytorch中Tensor的用法。

上一篇文章当中我们简单介绍了一下如何创建一个Tensor,今天我们继续深入Tensor的其他用法


tensor操作


size()和shape

我们可以用size()函数或者直接调用tensor当中的shape属性获取一个tensor的大小,这两者是等价的,一般情况下我们用前者多一些。


view

我们可以通过view改变一个tensor的shape,它会根据我们指定的shape返回一个新的tensor

需要注意的是,view返回的是原数据的一个引用,也就是说我们改变原数据,view出来的结果会同样发生变化。

在上面这个例子当中,我们把原tensor x中的[0, 1]的位置修改成了2,我们print y会发现y当中的元素同样发生了变化。


numel

我们可以用numel获取tenosr当中元素的数量:


squeeze和unsqueeze


我们可以用squeeze来减少tensor的维度,而使用unsqueeze来增加一个tenor的维度

其中unsqueeze接收一个参数,允许我们指定我们希望增加的维度。我们可以通

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
PyTorch是一个用于科学计算的开源机器学习库。以下是一些PyTorch的基础语法: 1. 创建张量: 通过torch.tensor()函数创建张量,可以指定张量的形状和数据类型。例如,torch.tensor([2.,3.2])创建了一个形状为的浮点张量。torch.Tensor()函数也可以用于创建张量,可以指定形状作为参数。例如,torch.Tensor(2,3)创建了一个形状为[2,3]的张量。默认情况下,PyTorch的张量类型是torch.FloatTensor。 2. 改变默认类型: 使用torch.set_default_tensor_type()函数可以改变默认张量类型。例如,torch.set_default_tensor_type(torch.DoubleTensor)可以将默认类型改为torch.DoubleTensor。 3. 初始化张量: 使用torch.rand()函数可以创建一个具有随机分布的张量。例如,a1=torch.rand(4,3,16,32)创建了一个形状为[4,3,16,32]的张量。 4. 张量的拼接: 使用torch.cat()函数可以将两个张量沿指定维度拼接在一起。例如,torch.cat([a1,a2], dim=2)可以将a1和a2沿第2个维度拼接在一起,得到形状为[4,3,32,32]的张量。另外,使用torch.stack()函数也可以实现拼接,不同之处在于它会额外创建一个新的维度。例如,torch.stack([a1,a2], dim=2)得到形状为[4,3,2,16,32]的张量。 5. 张量的分割: 使用torch.split()函数可以将一个张量沿指定维度分割成多个子张量。例如,torch.split(b, 2)可以将长度为2的张量b分割成两个子张量。 这些是PyTorch的一些基础语法,可以帮助你开始使用PyTorch进行科学计算机器学习

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值