Skip to main content

ClikaRT::ops::adaptive_avg_pool2d

function

adaptive_avg_pool2d()​

Tensor adaptive_avg_pool2d(Tensor x, ClikaRT::Span<const std::int64_t> output_size)

2-D adaptive average pooling of [N, H, W, C] to [N, H', W', C]; windows derived so the output lands exactly on output_size = {H', W'}. {1, 1} is global average pooling.

Throws

  • ClikaRT::Error: (INVALID_ARGUMENT) on a rank mismatch.

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