nn.adaptiveavgpool2d((1,1))
时间: 2023-05-02 14:05:07 浏览: 77
pytorch torch.nn.AdaptiveAvgPool2d()自适应平均池化函数详解
这是一个PyTorch中的函数,用来进行自适应平均池化操作。它将输入张量的每个通道上的特征图进行平均池化,并输出一个大小为(1,1)的张量。这个函数的主要目的是减小特征图的尺寸,以降低后续卷积层中的计算量。
阅读全文
pytorch torch.nn.AdaptiveAvgPool2d()自适应平均池化函数详解