通道数位于第1维

dim=1

（N, C, H, W）

posted @ 2020-09-16 20:30  6+0  阅读(135)  评论(0)    收藏  举报