- (beta) Channels Last Memory Format in PyTorch — PyTorch Tutorials 1.10.1+cu102 documentation
- Why does pytorch prefer using NCHW? - PyTorch Forums
すぐに忘れてしまうが、PyTorch は NCHW (BCHW) フォーマットを使っている・・・らしい。その意味では、https://github.com/yusugomori/deeplearning-keras-tf2-torch/blob/08e7b1fbd965eb5d6dbc50fd79c22dc4560ee5ea/6/utils/torch/DataLoader.py#L11 はちょっと気にはなるんだよなぁ・・・。