理解PyTorch维度概念 · x = torch.tensor([ · [1,2,3], · [4,5,6] · ]) · # 我们可以看到"行"是dim=0, "列"是dim=1 · print(x.shape) · >> torch.Size([2, 3]).
確定! 回上一頁