torch.nn.functional.adaptive_avg_pool2d

torch.nn.functional.adaptive_avg_pool2d(input, output_size) [source]

Применяет 2D адаптивную среднюю пуллинг к входному сигналу, состоящему из нескольких входных плоскостей.

См. AdaptiveAvgPool2d для подробностей и формы вывода.

Параметры

output_size (None) – целевой размер вывода (целое число или кортеж из двух целых чисел)

Тип возвращаемого значения

Тензор