1. ホーム
  2. python

[解決済み】torch.stack関数の使用方法

2022-01-30 22:58:55

質問

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

a.shape=(2, 3, 4) と b.shape=(2, 3) の2つのテンソルがあります。 スタックする方法 インプレース操作なしで?

どのように解決するのですか?

積み重ねには同じ次元数が必要です。一つの方法は、絞りを解除して積み重ねることです。例えば

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