oneflow.nn.functional.adaptive_avg_pool3d¶
-
oneflow.nn.functional.
adaptive_avg_pool3d
(input, output_size) → Tensor¶ Applies a 3D adaptive average pooling over an input signal composed of several input planes.
See
AdaptiveAvgPool3d
for details and output shape.- Parameters
input – the input tensor
output_size – the target output size (single integer or triple-integer tuple)
For examples:
>>> import oneflow as flow >>> import numpy as np >>> input = flow.tensor(np.random.randn(1, 1, 4, 4, 4), dtype=flow.float32) >>> output = flow.nn.functional.adaptive_avg_pool3d(input, (2, 2, 2))