Program Listing for File pixelshuffle.h

↰ Return to documentation for file (torch/csrc/api/include/torch/nn/options/pixelshuffle.h)

#pragma once

#include <torch/arg.h>
#include <torch/csrc/Export.h>
#include <torch/types.h>

namespace torch {
namespace nn {

struct TORCH_API PixelShuffleOptions {
  PixelShuffleOptions(int64_t upscale_factor)
      : upscale_factor_(upscale_factor) {}

  TORCH_ARG(int64_t, upscale_factor);
};

struct TORCH_API PixelUnshuffleOptions {
  /* implicit */ PixelUnshuffleOptions(int64_t downscale_factor)
      : downscale_factor_(downscale_factor) {}

  TORCH_ARG(int64_t, downscale_factor);
};

namespace functional {
using PixelShuffleFuncOptions = PixelShuffleOptions;

using PixelUnshuffleFuncOptions = PixelUnshuffleOptions;
} // namespace functional

} // namespace nn
} // namespace torch