“Comment faire la chaîne d'abord en pytorch” Réponses codées

Comment faire la chaîne d'abord en pytorch

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