nn_adaptive_avg_pool1d: Applies a 1D adaptive average pooling over an input signal...

In torch: Tensors and Neural Networks with 'GPU' Acceleration

Applies a 1D adaptive average pooling over an input signal composed of several input planes.

Description

The output size is H, for any input size. The number of output features is equal to the number of input planes.

Usage

nn_adaptive_avg_pool1d(output_size)

Arguments

output_size

the target output size H

Examples

if (torch_is_installed()) {
# target output size of 5
m <- nn_adaptive_avg_pool1d(5)
input <- torch_randn(1, 64, 8)
output <- m(input)
}

torch documentation built on May 29, 2024, 9:54 a.m.