ClikaRT::nn::Unflatten

class

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

Splits the dimension dim into sizes, whose product must equal its extent; a view when the layout permits. The module form of ClikaRT::ops::unflatten, which documents the full contract (shapes, dtypes, errors); construct once, then call it like a function.

Member functions

Unflatten(UnflattenOptions)

explicit Unflatten(UnflattenOptions options)

Constructs the module from options.

Declared in ClikaRT/nn/generated/flatten.h, line 133

Unflatten(int64_t, vector<int64_t>)

explicit Unflatten(std::int64_t dim, std::vector<std::int64_t> sizes)

Unflatten(dim, sizes): the positional form of UnflattenOptions(dim, sizes).

Declared in ClikaRT/nn/generated/flatten.h, line 136

forward()

Tensor forward(Tensor input) const

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

Declared in ClikaRT/nn/generated/flatten.h, line 148

operator()()

Tensor operator()(Tensor input) const

Same as forward.

Declared in ClikaRT/nn/generated/flatten.h, line 152

options()

const UnflattenOptions& options() const

The settings this module was constructed with.

Declared in ClikaRT/nn/generated/flatten.h, line 156