如何在Torch中复制矩阵行

时间:2016-04-10 18:58:32

标签: matrix lua torch

使用Torch,我将如何改变这个矩阵:

[1 2 3
 4 5 6
 7 8 9]

进入这个:

[1 2 3
 1 2 3
 1 2 3
 4 5 6
 4 5 6
 4 5 6
 7 8 9
 7 8 9
 7 8 9]

在Matlab中,我会这样做:

a = [1 2 3; 4 5 6; 7 8 9];
rows = repmat(1:size(a, 1), 1, 3);
b = a(rows(:), :);

Torch中有类似的方法吗?

1 个答案:

答案 0 :(得分:1)

这样可行:

a = a:repeatTensor(1, 3):reshape(9, 3)