Hello,
In this case, you would do [:, channel_number, :, :].
Actually, maybe I am not understanding your question fully. But if you have some number of channels with a certain height and width, I don’t think you can reshape to 1D with just (channel,). Could you please elaborate more perhaps?