ClikaRT::nn::PixelUnshuffle

class

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

Moves spatial samples into the channels: [N, H * r, W * r, C] to [N, H, W, C * r * r] for the factor r. The module form of ClikaRT::ops::pixel_unshuffle, which documents the full contract (shapes, dtypes, errors); construct once, then call it like a function.

Member functions

PixelUnshuffle(PixelUnshuffleOptions)

explicit PixelUnshuffle(PixelUnshuffleOptions options)

Constructs the module from options.

Declared in ClikaRT/nn/generated/pixelshuffle.h, line 130

PixelUnshuffle(int64_t)

explicit PixelUnshuffle(std::int64_t downscale_factor)

PixelUnshuffle(downscale_factor): the positional form of PixelUnshuffleOptions(downscale_factor).

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

forward()

Tensor forward(Tensor input) const

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

Declared in ClikaRT/nn/generated/pixelshuffle.h, line 146

operator()()

Tensor operator()(Tensor input) const

Same as forward.

Declared in ClikaRT/nn/generated/pixelshuffle.h, line 150

options()

const PixelUnshuffleOptions& options() const

The settings this module was constructed with.

Declared in ClikaRT/nn/generated/pixelshuffle.h, line 154