Hello there,
I am using following code in model ,
I=torch.randn(40,1,128,186)
m=nn.Sequential(
nn.Conv2d(1,16, kernel_size=5, stride=1),
nn.BatchNorm2d(16),
nn.ReLU())
O=m(I)
I want my output to be of same size as input ie (40,1,128,186)
I tried using
O=m(F.pad(I,(0,0,2,2)))
But getting (40, 16, 128, 182)
Can anyone help plz.