nnf_avg_pool1d: Avg_pool1d

View source: R/nnf-pooling.R

nnf_avg_pool1dR Documentation

Avg_pool1d

Description

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

Usage

nnf_avg_pool1d(
  input,
  kernel_size,
  stride = NULL,
  padding = 0,
  ceil_mode = FALSE,
  count_include_pad = TRUE
)

Arguments

input

input tensor of shape (minibatch , in_channels , iW)

kernel_size

the size of the window. Can be a single number or a tuple ⁠(kW,)⁠.

stride

the stride of the window. Can be a single number or a tuple ⁠(sW,)⁠. Default: kernel_size

padding

implicit zero paddings on both sides of the input. Can be a single number or a tuple ⁠(padW,)⁠. Default: 0

ceil_mode

when True, will use ceil instead of floor to compute the output shape. Default: FALSE

count_include_pad

when True, will include the zero-padding in the averaging calculation. Default: TRUE


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