Pytorch 和Tensorflow之输入图片数据的data_forma区别

Pytorch默认是 NCHW
Tensorflow默认是NHWC

N - batchsize
H - 图片高
W - 图片宽
C - 通道数

https://blog.csdn.net/edward_zcl/article/details/102661617
https://www.sohu.com/a/336743474_473283

原创文章 96 获赞 24 访问量 3万+

猜你喜欢

转载自blog.csdn.net/c2250645962/article/details/105221217