pytorchのconcatの方法が知りたいです

Question

以下のコードはtensorflowのt1とt2のテンソルをconcatするコードなのですが，
axisを-1にしてconcatしています．
以下のコードをpytorchに書き換える場合はどのようにしたら良いでしょうか？
(axisを-1にしてconcatしたいです．)

tf.concat([t1, t2], -1)

Answer 1

torch.cat() でできます。

import torch

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

c = torch.cat([a, b], axis=0)
print(c)
# tensor([[-0.2491, -0.2652, -0.2173],
#         [-0.5421, -0.9373, -0.4611],
#         [-0.0446, -1.4220, -0.5104],
#         [-2.5595, -0.2290,  0.3934],
#         [ 0.4208,  2.8840, -1.1338],
#         [-0.5057,  0.8306,  1.1929]])

c = torch.cat([a, b], axis=1)
print(c)
# tensor([[-0.2491, -0.2652, -0.2173, -2.5595, -0.2290,  0.3934],
#         [-0.5421, -0.9373, -0.4611,  0.4208,  2.8840, -1.1338],
#         [-0.0446, -1.4220, -0.5104, -0.5057,  0.8306,  1.1929]])