Hi, I got sing channel images from ‘.mat’. And I write my custom Dataset, but got some errors.RuntimeError: output with shape [1, 21, 21] doesn’t match the broadcast shape [3, 21, 21]

my images array shape(1596, 21, 21)