ClikaRT::ops::max_pool_with_indices

function

max_pool_with_indices()

std::array<Tensor 2> max_pool_with_indices(
    Tensor x,
    ClikaRT::Span<const std::int64_t> kernel_size,
    ClikaRT::Span<const std::int64_t> stride,
    ClikaRT::Span<const std::int64_t> padding,
    ClikaRT::Span<const std::int64_t> dilation,
    bool ceil_mode
)

Returns: a 2-element array {values, indices}.

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