在 Torch 中组合张量的最惯用的方法是什么？

Question

我遇到了将三个张量连接在一起的问题，这样 3 px1 张量就变成了一个 3px1 张量。

我能想到的最简洁的是：

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

有没有办法在不转换为表格并返回张量的情况下做到这一点？假设它们具有兼容的形状，似乎应该有一种通用的方法可以沿着某个指定的维度连接张量。

我明白怎么可能写出这样一个函数，不存在吗？

Answer 1

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)

What are the most idiomatic ways to combine tensors in torch?