Program Listing for File pixelshuffle.h¶
↰ Return to documentation for file (torch/csrc/api/include/torch/nn/modules/pixelshuffle.h
)
#pragma once
#include <torch/nn/cloneable.h>
#include <torch/nn/functional/pixelshuffle.h>
#include <torch/nn/options/pixelshuffle.h>
#include <torch/csrc/Export.h>
namespace torch::nn {
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ PixelShuffle
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
struct TORCH_API PixelShuffleImpl
: public torch::nn::Cloneable<PixelShuffleImpl> {
explicit PixelShuffleImpl(const PixelShuffleOptions& options_);
void pretty_print(std::ostream& stream) const override;
Tensor forward(const Tensor& input);
void reset() override;
PixelShuffleOptions options;
};
TORCH_MODULE(PixelShuffle);
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ PixelUnshuffle ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
struct TORCH_API PixelUnshuffleImpl
: public torch::nn::Cloneable<PixelUnshuffleImpl> {
explicit PixelUnshuffleImpl(const PixelUnshuffleOptions& options_);
void pretty_print(std::ostream& stream) const override;
Tensor forward(const Tensor& input);
void reset() override;
PixelUnshuffleOptions options;
};
TORCH_MODULE(PixelUnshuffle);
} // namespace torch::nn