Comment utiliser la fonction torch.stack
J'ai une question sur torch.stack
J'ai 2 tenseurs, a.shape = (2, 3, 4) et b.shape = (2, 3). Comment les empiler sans opération sur place?
L'empilement nécessite le même nombre de dimensions. Une façon serait de décompresser et d'empiler. Par exemple:
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