举个简单的例子:输入的是(1,1, 3,3)大小的tensor,然后在 nn.AdaptiveAvgPool2d(2)的作用下,会成为一个2*2的tensor,首先要知道采用的是平均池化,比如:
1,2,3
4,5,6
7,8,9
因为函数是自适应,所以strides,paddings等参数都自适应好了,这个例子中,strides=1,最终结果为:
3,4
6,7
这是我的理解,有错误欢迎指正,谢谢
本文通过一个简单示例,详细解释了PyTorch中AdaptiveAvgPool2d函数的工作原理。输入一个(1,1,3,3)大小的张量,在经过AdaptiveAvgPool2d(2)处理后,输出变为2*2的张量。文章深入探讨了平均池化过程及自适应调整的strides和paddings参数。
举个简单的例子:输入的是(1,1, 3,3)大小的tensor,然后在 nn.AdaptiveAvgPool2d(2)的作用下,会成为一个2*2的tensor,首先要知道采用的是平均池化,比如:
1,2,3
4,5,6
7,8,9
因为函数是自适应,所以strides,paddings等参数都自适应好了,这个例子中,strides=1,最终结果为:
3,4
6,7
这是我的理解,有错误欢迎指正,谢谢
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
619
1863
1088

被折叠的 条评论
为什么被折叠?