Tensorflow: [batch, height, width, channel]
Pytorch: [batch, channel, height, width]
Tensorflow 和 Pytorch 的张量表示
最新推荐文章于 2022-11-21 19:22:54 发布
Tensorflow: [batch, height, width, channel]
Pytorch: [batch, channel, height, width]