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

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

N：batch

C：channel

H：height

W：width

posted @ 2021-03-16 14:28  一剑光寒十四州  阅读(276)  评论(0)  编辑  收藏  举报