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

最新推荐文章于 2024-06-06 14:26:34 发布

rrr2

最新推荐文章于 2024-06-06 14:26:34 发布

阅读量5.9k

点赞数 1

分类专栏： torch pytorch学习

本文链接：https://blog.csdn.net/qq_35608277/article/details/85872296

版权

torch 同时被 2 个专栏收录

49 篇文章 1 订阅

订阅专栏

pytorch学习

42 篇文章 2 订阅

订阅专栏

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

只需要输出张量，
卷积核，可以一个数可以不是方形的。

$Parameters: input – input tensor (\text{minibatch} \times \text{in_channels} \times iH \times iW)(minibatch×in_channels×iH×iW)kernel_size – size of the pooling region. Can be a single number or a tuple (kH \times kW)(kH×kW)stride – stride of the pooling operation. Can be a single number or a tuple (sH, sW). Default: kernel_sizepadding – implicit zero paddings on both sides of the input. Can be a single number or a tuple (padH, padW). Default: 0ceil_mode – when True, will use ceil instead of floor in the formula to compute the output shape. Default: Falsecount_include_pad – when True, will include the zero-padding in the averaging calculation. Default: True$

尺寸
在这里插入图片描述

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

rrr2

关注

1
点赞
踩
5

收藏

觉得还不错? 一键收藏
0
评论
torch.nn.functional.avg_pool2d(input, kernel_size, stride=None, padding=0, ceil_mode=False, count_in

refhttps://pytorch.org/docs/stable/nn.html#torch-nn-functionaltorch.nn.functional.avg_pool2d(input, kernel_size, stride=None, padding=0, ceil_mode=False, count_include_pad=True) → Tensor只需要输出张量，...
复制链接

扫一扫