torch.nn.functional.adaptive_avg_pool2d

torch.nn.functional.adaptive_avg_pool2d(input, output_size)

對由多個輸入通道組成的輸入訊號應用二維自適應平均池化。

有關詳細資訊和輸出形狀，請參閱 AdaptiveAvgPool2d。

引數

output_size (None) – 目標輸出尺寸（單個整數或雙整數元組）

返回型別

張量