Assume that a conv layer’s output is bchw, you can set the last channel is 6, and using global pooling by grouped to change output from hw to h1 or w1, then you do some dimensionality reduction. you will get the bX6©.