Tensor——各个机器学习框架下Tensor中的通道顺序(一)
| 序号 | 框架 | 通道顺序 |
| 1 | Caffe | NCHW |
| 2 | Tensorflow | 默认NHWC,也支持NCHW |
| 3 | PyTorch | NCHW |
| 4 | TensorRT | NCHW |
N:batch
C:channel
H:height
W:width
| 序号 | 框架 | 通道顺序 |
| 1 | Caffe | NCHW |
| 2 | Tensorflow | 默认NHWC,也支持NCHW |
| 3 | PyTorch | NCHW |
| 4 | TensorRT | NCHW |
N:batch
C:channel
H:height
W:width