Stacking requires same number of dimensions. One way would be to unsqueeze and stack. For example: a.size() # 2, 3, 4 b.size() # 2, 3 b = torch.unsqueeze(b, ...
確定! 回上一頁