HWC 转换 CHW

a.shape()    #(100,100,3)

b=a.transpose(3, 1, 2)

b.shape()   #(3,100,100)

posted @ 2020-06-20 10:44  落地成霜  阅读(1760)  评论(0)  编辑  收藏  举报