通道数位于第1维

dim=1

(N, C, H, W)

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