nnf_adaptive_avg_pool1d: Adaptive_avg_pool1d

View source: R/nnf-pooling.R

nnf_adaptive_avg_pool1dR Documentation

Adaptive_avg_pool1d

Description

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

Usage

nnf_adaptive_avg_pool1d(input, output_size)

Arguments

input

input tensor of shape (minibatch , in_channels , iW)

output_size

the target output size (single integer)


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