torch.stack関数の使用方法

Question

Torch.stackについて質問があります

2つのテンソル、a.shape =（2、3、4）とb.shape =（2、3）があります。 それらをスタックする方法インプレース操作なし？

Answer 1

積み重ねには同じ数の次元が必要です。 1つの方法は、圧縮解除してスタックすることです。例えば：

a.size()  # 2, 3, 4
b.size()  # 2, 3
b = torch.unsqueeze(b, dim=2)  # 2, 3, 1
# torch.unsqueeze(b, dim=-1) does the same thing

torch.stack([a, b], dim=2)  # 2, 3, 5