Tensor——各个机器学习框架下Tensor中的通道顺序(一)

序号 框架 通道顺序
1 Caffe NCHW
2 Tensorflow 默认NHWC,也支持NCHW
3 PyTorch NCHW
4 TensorRT NCHW

N:batch

C:channel

H:height

W:width

原文地址:https://www.cnblogs.com/timelesszxl/p/14543326.html