pytorch AdaptivePool2d

903 ワード

参照リンク:https://pytorch.org/docs class torch.nn. AdaptiveAvgPool2d (output_size)
2 Dアダプティブ平均プール(2 D adaptive average pooling)は、複数の入力平面からなる入力信号に適用される.
入力サイズのいずれについても、出力のサイズはhxwです.出力フィーチャーの数は、入力プレーンの数に等しい.
パラメータ:
output_size:目標出力サイズの画像の形式H×W.メタセット(H,W)または正方形画像H x Hの個別のHであってもよい.HとWはintであってもよく、Noneであってもよい.これはsizeがinputと同じであることを意味する.
>>> # target output size of 5x7
>>> m = nn.AdaptiveAvgPool2d((5,7))
>>> input = torch.randn(1, 64, 8, 9)
>>> output = m(input)
>>> # target output size of 7x7 (square)
>>> m = nn.AdaptiveAvgPool2d(7)
>>> input = torch.randn(1, 64, 10, 9)
>>> output = m(input)
>>> # target output size of 10x7
>>> m = nn.AdaptiveMaxPool2d((None, 7))
>>> input = torch.randn(1, 64, 10, 9)
>>> output = m(input)