Pytorch Conv layers expect the shape to be [B, C, H, W](in ur case: [5384, 3, 64, 64]); however, ur shape is [H, W, C, B](64, 64, 3, 5384), and ur original shape should be [B, H, W, C]([5384, 64, 64, 3]).