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