I am confused on how to replicate keras (tensorflow) convolution to PyTorch.
In keras, I can do something like this. (the input size is (256,237,1,21) and the output
