Based on the error message it seems that you are trying to reshape PW_conv4 to a shape of [2, 64, 50, 50], which is invalid given that PW_conv4 contains 48384 elements.
Could you explain how you’ve calculated the shape for the reshape operation and then check where the size mismatch is coming from?