在火炬中结合张量的最惯用方法是什么?

时间:2015-07-02 13:58:36

标签: lua torch

我面临着将三个张量连接在一起,因此3个px1张量变为一个3px1张量。

我能想到的最简洁的是:

torch.Tensor{v2:totable(),v4:totable(),v6:totable()}:view(3*p,1)

有没有办法在不转换为表格并返回张量的情况下执行此操作?似乎应该有一种将张量连接到某个指定维度的通用方法,假设它们具有兼容的形状。

我可以看到如何编写这样的函数,不存在吗?

1 个答案:

答案 0 :(得分:16)

a = torch.randn(3,1)
b = torch.randn(3,1)
c = torch.randn(3,1)

d = torch.cat(a,b,1):cat(c,1)

print(d)