Tensorflow 和 Pytorch 的张量表示

Tensorflow: [batch, height, width, channel]
Pytorch: [batch, channel, height, width]

发布了108 篇原创文章 · 获赞 7 · 访问量 4379

猜你喜欢

转载自blog.csdn.net/weixin_44795555/article/details/104812640