Struct UnfoldOptions

• Defined in File fold.h

Struct Documentation

struct UnfoldOptions

Options for the Unfold module.

Example:

Unfold model(UnfoldOptions({2, 4}).dilation(2).padding({2, 1}).stride(2));

Public Functions

inline UnfoldOptions(ExpandingArray<2> kernel_size)

inline auto kernel_size(const ExpandingArray<2> &new_kernel_size) -> decltype(*this)

the size of the sliding blocks

inline auto kernel_size(ExpandingArray<2> &&new_kernel_size) -> decltype(*this)

inline const ExpandingArray<2> &kernel_size() const noexcept

inline ExpandingArray<2> &kernel_size() noexcept

inline auto dilation(const ExpandingArray<2> &new_dilation) -> decltype(*this)

controls the spacing between the kernel points; also known as the à trous algorithm.

inline auto dilation(ExpandingArray<2> &&new_dilation) -> decltype(*this)

inline const ExpandingArray<2> &dilation() const noexcept

inline ExpandingArray<2> &dilation() noexcept

inline auto padding(const ExpandingArray<2> &new_padding) -> decltype(*this)

controls the amount of implicit zero-paddings on both sides for padding number of points for each dimension before reshaping.

inline auto padding(ExpandingArray<2> &&new_padding) -> decltype(*this)

inline const ExpandingArray<2> &padding() const noexcept

inline ExpandingArray<2> &padding() noexcept

inline auto stride(const ExpandingArray<2> &new_stride) -> decltype(*this)

controls the stride for the sliding blocks.

inline auto stride(ExpandingArray<2> &&new_stride) -> decltype(*this)

inline const ExpandingArray<2> &stride() const noexcept

inline ExpandingArray<2> &stride() noexcept