Shortcuts

Struct PixelShuffleImpl

Inheritance Relationships

Base Type

Struct Documentation

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

Rearranges elements in a tensor of shape :math:(*, C \times r^2, H, W) to a tensor of shape :math:(*, C, H \times r, W \times r), where r is an upscale factor.

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

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

Example:

PixelShuffle model(PixelShuffleOptions(5));

Public Functions

PixelShuffleImpl(const PixelShuffleOptions &options_)
void pretty_print(std::ostream &stream) const override

Pretty prints the PixelShuffle module into the given stream.

Tensor forward(const Tensor &input)
void reset() override

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

Public Members

PixelShuffleOptions options

The options with which this Module was constructed.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources