i see i will try the change you suggested! it makes much more sense adding the fourth dimension, the four dimensions are batch, channel (50 because the input is the result of a previous convolution layer) and then height, width