ClikaRT::ops::max_pool_with_indices

function

max_pool_with_indices()

std::array<Tensor 2> max_pool_with_indices(
    Tensor input,
    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
)

Max pooling that also returns each window's winning element as a flat index over the input's spatial extents (per plane, the same for every channel). The values follow max_pool. The index names the window's lowest-index NaN element when it holds one, else the first element holding the max (a tie keeps the lowest index); a window lying wholly in the padding reports -1.

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

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