PyTorch Forums
Swap axes in pytorch?
apaszke
(Adam Paszke)
March 9, 2017, 1:14pm
2
No, we only support
NCHW
format. You can use
.permute
to swap the axis.
22 Likes
show post in topic