函数原型
torch.nn.functional.conv2d(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1)
参数:
这里仅仅解释两个主要的参数。
以一个例子说明:
# With square kernels and equal stride
filters = torch.randn(8, 4, 3, 3)
inputs = torch.randn(1, 4, 5, 5)
out=F.conv2d(inputs, filters)
out的shape为(1,8,3,3)
注意input和weight的参数含义,就能清楚了。