torch.nn.functional.avg_pool2d(input, kernel_size, stride=None, padding=0, ceil_mode=False, count_in

ref

https://pytorch.org/docs/stable/nn.html#torch-nn-functional

torch.nn.functional.avg_pool2d(input, kernel_size, stride=None, padding=0, ceil_mode=False, count_include_pad=True) → Tensor

隻需要輸出張量，

卷積核，可以一個數可以不是方形的。

尺寸

inputx= torch.randn(2, 3, 4, 4)
F.avg_pool2d(inputx,2,1)