PyTorch中permute的用法 根据指定的维度进行转置

pytorch 中的permute 就像是numpy中的transpose()函数一样, 根据指定的维度进行转置, 我们这里跟view()进行一下对比

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
PyTorch的扩张维度操作可以使用`expand()`或`repeat()`函数实现。`expand()`函数将指定维度扩展为相应的维度大小,基于原来的数据。要注意的是,要扩展的维度必须是1维的。而`repeat()`函数则是将整个张量在指定维度进行复制。下面是一个示例: 对于扩张维度操作,可以使用`expand()`函数。例如,对于张量`a`的形状为`(1,32,1,1)`,可以通过`a.expand(4,32,14,14)`将第一个维度扩展为4,得到新的形状为`(4,32,14,14)`的张量。 而对于复制维度操作,可以使用`repeat()`函数。例如,对于张量`a`的形状为`(1,32,1,1)`,可以通过`a.repeat(4,32,14,14)`将整个张量在指定维度进行复制,得到新的形状为`(4,1024,14,14)`的张量。 另外,还可以使用`permute()`函数和`transpose()`函数来交换张量的维度顺序。`permute()`函数根据参数维度的索引,将对应维度放在指定的位置上。`transpose()`函数则是将指定维度进行转置。下面是一个示例: 对于交换维度顺序,可以使用`permute()`函数。例如,对于形状为`(4,3,28,32)`的张量`a`,可以通过`a.permute(1,3,2,0)`将第1个维度放在第0个位置,第3个维度放在第1个位置,第2个维度放在第2个位置,第0个维度放在第3个位置,得到新的形状为`(3,32,28,4)`的张量。 也可以使用`transpose()`函数来转置维度。例如,对于形状为`(4,3,28,32)`的张量`a`,可以通过`a.transpose(1,3)`将第1个维度和第3个维度进行转置,得到新的形状为`(4,32,28,3)`的张量。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值