Skip to main content

ClikaRT::ops::avg_pool1d

function

avg_pool1d()

Tensor avg_pool1d(
    Tensor x,
    ClikaRT::Span<const std::int64_t> kernel_size,
    ClikaRT::Span<const std::int64_t> stride = {},
    ClikaRT::Span<const std::int64_t> padding = {0},
    bool ceil_mode = false,
    bool count_include_pad = true
)

1-D average pooling over [N, L, C] (channels-last). See avg_pool; stride empty = kernel_size.

Returns: [N, L', C], dtype of x.

Throws

  • ClikaRT::Error: (INVALID_ARGUMENT) on mismatched parameter ranks.

Declared in ClikaRT/compute/ops.h, line 3898