🌈 搜索
📅  最后修改于: 2022-03-11 15:00:27.696000             🧑  作者: Mango
X_train = torch.rand(708, 256, 3) X_train = X_train.permute(2, 0, 1) X_train.shape # => torch.Size([3, 708, 256])