ClikaRT::nn::AvgPool2d

class

Header: ClikaRT/nn/generated/pooling.h
Inherits: ClikaRT::nn::Module

Average pooling over a 2-D window; the input is channels-last [N, D1..D2, C]. The module form of ClikaRT::ops::avg_pool2d, which documents the full contract (shapes, dtypes, errors); construct once, then call it like a function.

Member functions

AvgPool2d(AvgPool2dOptions)

explicit AvgPool2d(AvgPool2dOptions options)

Constructs the module from options.

Declared in ClikaRT/nn/generated/pooling.h, line 515

AvgPool2d(vector<int64_t>)

explicit AvgPool2d(std::vector<std::int64_t> kernel_size)

AvgPool2d(kernel_size): the positional form of AvgPool2dOptions(kernel_size).

Declared in ClikaRT/nn/generated/pooling.h, line 518

forward()

Tensor forward(Tensor input) const

forward_impl, unwrapped: raises ClikaRT::Error on failure.

Declared in ClikaRT/nn/generated/pooling.h, line 533

operator()()

Tensor operator()(Tensor input) const

Same as forward.

Declared in ClikaRT/nn/generated/pooling.h, line 537

options()

const AvgPool2dOptions& options() const

The settings this module was constructed with.

Declared in ClikaRT/nn/generated/pooling.h, line 541