Answers for "pytorch channel first to channel last"

1

how to do channel first in pytorch

X_train = torch.rand(708, 256, 3)
X_train = X_train.permute(2, 0, 1)
X_train.shape
# => torch.Size([3, 708, 256])
Posted by: Guest on April-23-2021

Code answers related to "pytorch channel first to channel last"

Browse Popular Code Answers by Language