ClikaRT::ops::max_pool3d

function

max_pool3d()

Tensor max_pool3d(
    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, 0, 0},
    ClikaRT::Span<const std::int64_t> dilation = {1, 1, 1},
    bool ceil_mode = false
)

3-D max pooling over [N, D, H, W, C] (channels-last). See max_pool2d; parameters extend to {kD, kH, kW} etc.

Returns: [N, D', H', W', C], dtype of x.

Throws

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

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