您的位置:首页 > 编程语言 > Lua

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

2015-11-03 16:13 411 查看
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]
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  lua 语言