Do you have 16 images in a batch, where each sample is associated with a single class?
Or does each image have 16 channels and you would like to split it somehow?
Could you explain your use case and post some dummy code which shows how your model should work?
Generally, nn.Conv2d expects an input of [batch_size, channels, height, width].