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

2024-02-23 10:00:40

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

N：batch

C：channel

H：height

W：width

上一篇：2019，英伟达的低谷与涅槃


下一篇：mmpose PyTorch模型转TensorRT