nn.adaptiveavgpool2d((1, 1))
时间: 2023-04-25 21:04:23 浏览: 458
pytorch torch.nn.AdaptiveAvgPool2d()自适应平均池化函数详解
nn.adaptiveavgpool2d((1, 1)) 是一个 PyTorch 中的函数,它表示自适应平均池化层,其输入是一个二维张量,输出是一个大小为 (1, 1) 的二维张量。自适应平均池化层可以根据输入的大小自动调整池化层的大小,从而适应不同大小的输入。
阅读全文
pytorch torch.nn.AdaptiveAvgPool2d()自适应平均池化函数详解