Comment Comment by apl You can almost 1:1 translate this by swapping "tf" and "torch". No need to use nn.Conv2d -- there's a functional API for all these layers:https://pytorch.org/docs/master/nn.functional.html#conv2dTorch doesn't have "same" padding, so you have to manually calculate the correct padding value for your input/output shapes. parent apl 6 years ago
https://pytorch.org/docs/master/nn.functional.html#conv2d
Torch doesn't have "same" padding, so you have to manually calculate the correct padding value for your input/output shapes.