Struct UnfoldOptions#

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#