Learn R Programming

torch (version 0.0.2)

torch_avg_pool1d: Avg_pool1d

Description

Avg_pool1d

Arguments

input

NA input tensor of shape \((\mbox{minibatch} , \mbox{in\_channels} , iW)\)

kernel_size

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

stride

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

padding

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

ceil_mode

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

count_include_pad

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

avg_pool1d(input, kernel_size, stride=None, padding=0, ceil_mode=False, count_include_pad=True) -> Tensor

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

See ~torch.nn.AvgPool1d for details and output shape.