ClikaRT::ops::avg_pool3d

function

avg_pool3d()

Tensor avg_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},
    bool ceil_mode = false,
    bool count_include_pad = true,
    std::optional<int64_t> divisor_override = std::nullopt
)

3-D average pooling over [N, D, H, W, C] (channels-last). See avg_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 3924