KarthikR:

import torch
import torch.nn as nn
input = torch.randn(512, 1, 28, 28)
print(input.shape)
conv1 = nn.Conv2d(1, 128, 5, 2, bias=False)
print(conv1(input).shape)


Thank you so much.
I could understand.
Also, Thank you for telling sample code, it was very useful for me.
Thank you.