Could I ask the stupid question, what’s the meaning of “different shapes (5 or 8).” ? For e.g., My input shape is (3,512,512)(chw) and the outpust shape is (80, 128,128), if some data get shape 5 or 8, which dimension of the data is 5 or 8