Why torch.nn.Conv2d(in_channels=3, out_channels=4, kernel_size=(3,2)) have a weight.shape of (4,3,3,2) but will output (4,3,3)?
my goal is to understand torch.nn.Conv2d
. Right now, I am confused with the kernel_size
.
my goal is to understand torch.nn.Conv2d
. Right now, I am confused with the kernel_size
.