使用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中有类似的方法吗?
答案 0 :(得分:1)
这样可行:
a = a:repeatTensor(1, 3):reshape(9, 3)