Struct UpsampleOptions

• Defined in File upsampling.h

Struct Documentation

struct UpsampleOptions

Options for the Upsample module.

Example:

Upsample
model(UpsampleOptions().scale_factor(std::vector<double>({3})).mode(torch::kLinear).align_corners(false));

Public Functions

inline auto size(const c10::optional<std::vector<int64_t>> &new_size) -> decltype(*this)

output spatial sizes.

inline auto size(c10::optional<std::vector<int64_t>> &&new_size) -> decltype(*this)

inline const c10::optional<std::vector<int64_t>> &size() const noexcept

inline c10::optional<std::vector<int64_t>> &size() noexcept

inline auto scale_factor(const c10::optional<std::vector<double>> &new_scale_factor) -> decltype(*this)

multiplier for spatial size.

inline auto scale_factor(c10::optional<std::vector<double>> &&new_scale_factor) -> decltype(*this)

inline const c10::optional<std::vector<double>> &scale_factor() const noexcept

inline c10::optional<std::vector<double>> &scale_factor() noexcept

inline auto mode(const mode_t &new_mode) -> decltype(*this)

inline auto mode(mode_t &&new_mode) -> decltype(*this)

inline const mode_t &mode() const noexcept

inline mode_t &mode() noexcept

inline auto align_corners(const c10::optional<bool> &new_align_corners) -> decltype(*this)

if “True”, the corner pixels of the input and output tensors are aligned, and thus preserving the values at those pixels.

This only has effect when :attr:mode is “linear”, “bilinear”, “bicubic”, or “trilinear”. Default: “False”

inline auto align_corners(c10::optional<bool> &&new_align_corners) -> decltype(*this)

inline const c10::optional<bool> &align_corners() const noexcept

inline c10::optional<bool> &align_corners() noexcept