📜  如何在 pytorch 中先做通道 - 任何代码示例

📅  最后修改于: 2022-03-11 15:00:27.696000             🧑  作者: Mango

代码示例1
X_train = torch.rand(708, 256, 3)
X_train = X_train.permute(2, 0, 1)
X_train.shape
# => torch.Size([3, 708, 256])