神经网络的特征图(B,C,W,H)如何按照通道数划分为三个特征图，求解答，非常感谢

Question

神经网络的特征图(B,C,W,H)如何按照通道数划分为三个特征图，求解答，非常感谢

Answer 1

1.最简单的方法，直接截
假设a.shape 为[b,c,w,h]，
b1=[b,:c//3,w,h]
b2=[b,c//3:c//32,w,h]
b3=[b,c//32:,w,h]
2.做卷积