Thanks for your input.
My input to the Conv2d layer was torch.Size([4, 1, 256, 768])
where 4 is the number of samples in a single batch. Why isn’t the output: [4,3,252,1] ?
(4 samples, 3 output channels, 252 height, 1 width>)