pytorch——permute,view

permute(),适合多维数据。可以灵活的对原数据的维度进行调换,而数据本身不变。

view() 把原先tensor中的数据按照行优先的顺序排成一个一维的数据(这里应该是因为要求地址是连续存储的),然后按照参数组合成其他维度的tensor。比如说是不管你原先的数据是[[[1,2,3],[4,5,6]]]还是[1,2,3,4,5,6],因为它们排成一维向量都是6个元素,所以只要view后面的参数一致,得到的结果都是一样的。

import torch
a = torch.tensor([[1,2,3],[4,5,6]])
print(a,a.size())
print(a.view(3,2))
print(a.permute(1,0))

b = torch.tensor([1,2,3,4,5,6])
# view之后的tensor不一定与之前的tensor的维度数一样
print(b.view(3,2))


tensor([[1, 2, 3],
        [4, 5, 6]]) torch.Size([2, 3])

tensor([[1, 2],
        [3, 4],
        [5, 6]])

tensor([[1, 4],
        [2, 5],
        [3, 6]])

tensor([[1, 2],
        [3, 4],
        [5, 6]])

permute函数传入的是维度,view函数传入的是维度上的维数。
 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值