Can we define a 2d convolution inpytorch in the following way:
pred_bb = nn.conv2d(x_BB,4, (net_shape[1], net_shape[2]))
x_BB is the output of previous convolutional layers
4 is the number of output channels
and, (net_shape[1], net_shape[2])  is the shape of the kernel
Thanks and regards