lua语言中transpose()函数的用法

transpose英文意思是‘变换顺序’的意思,在lua语言中,lua表示对矩阵或者张量进行维度交换。
b = a:transpose(m,n)表示将矩阵或者张量a的第m维和第n维交换。
若a是一个二维矩阵,m=1,n=2,则b = a:transpose(1,2)表示矩阵a的转置。
举例1

a = torch.rand(4,5)     print('矩阵a:\n',a)
b = a:transpose(1,2)    print('矩阵a的转置:\n',b)

结果如下:
矩阵a:
0.7638 0.7820 0.7317 0.0020 0.1337
0.0577 0.7359 0.7685 0.7482 0.4766
0.5441 0.3134 0.9738 0.8549 0.6960
0.3207 0.8908 0.8622 0.4824 0.7444
[torch.DoubleTensor of size 4x5]

矩阵a的转置:
0.7638 0.0577 0.5441 0.3207
0.7820 0.7359 0.3134 0.8908
0.7317 0.7685 0.9738 0.8622
0.0020 0.7482 0.8549 0.4824
0.1337 0.4766 0.6960 0.7444
[torch.DoubleTensor of size 5x4]
%———————————————————–%
举例2

a = torch.Tensor(3,4,5,6)   
print('转换前的维度:\n',#a)
b = a:transpose(2,4)       --表示将a的第2维和第4维进行交换       
print('转换后的各维大小:\n',#b)

结果如下:
转换前的维度:
3
4
5
6
[torch.LongStorage of size 4]
转换后的各维大小:
3
6
5
4
[torch.LongStorage of size 4]

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值