oneflow.nn.functional.adaptive_avg_pool1d¶
-
oneflow.nn.functional.
adaptive_avg_pool1d
(input, output_size) → Tensor¶ Applies a 1D adaptive average pooling over an input signal composed of several input planes.
See
AdaptiveAvgPool1d
for details and output shape.- Parameters
input – the input tensor
output_size – the target output size (single integer)
For examples:
>>> import oneflow as flow >>> import numpy as np >>> arr = np.array([[[ 0.0558, -0.6875, -1.6544, -0.6226, 0.1018, 0.0502, -1.2538, 0.1491]]]) >>> input = flow.tensor(arr, dtype=flow.float32) >>> flow.nn.functional.adaptive_avg_pool1d(input, output_size=[4]) tensor([[[-0.3158, -1.1385, 0.0760, -0.5524]]], dtype=oneflow.float32)