Struct PixelUnshuffleImpl

• Defined in File pixelshuffle.h

Inheritance Relationships

Base Type

• public torch::nn::Cloneable< PixelUnshuffleImpl > (Template Class Cloneable)

Struct Documentation

struct PixelUnshuffleImpl : public torch::nn::Cloneable<PixelUnshuffleImpl>

Reverses the PixelShuffle operation by rearranging elements in a tensor of shape :math:(*, C, H \times r, W \times r) to a tensor of shape :math:(*, C \times r^2, H, W), where r is a downscale factor.

See https://pytorch.org/docs/main/nn.html#torch.nn.PixelUnshuffle to learn about the exact behavior of this module.

See the documentation for torch::nn::PixelUnshuffleOptions class to learn what constructor arguments are supported for this module.

Example:

PixelUnshuffle model(PixelUnshuffleOptions(5));

Public Functions

explicit PixelUnshuffleImpl(const PixelUnshuffleOptions &options_)

virtual void pretty_print(std::ostream &stream) const override

Pretty prints the PixelUnshuffle module into the given stream.

Tensor forward(const Tensor &input)

virtual void reset() override

reset() must perform initialization of all members with reference semantics, most importantly parameters, buffers and submodules.

Public Members

PixelUnshuffleOptions options

The options with which this Module was constructed.