Template Struct MaxPoolOptions#

Struct Documentation#

template<size_t D>
struct MaxPoolOptions#

Options for a D-dimensional maxpool module.

Public Functions

inline MaxPoolOptions(ExpandingArray<D> kernel_size)#

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

the size of the window to take a max over

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

inline const ExpandingArray<D> &kernel_size() const noexcept#

inline ExpandingArray<D> &kernel_size() noexcept#

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

the stride of the window. Default value is `kernel_size

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

inline const ExpandingArray<D> &stride() const noexcept#

inline ExpandingArray<D> &stride() noexcept#

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

implicit zero padding to be added on both sides

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

inline const ExpandingArray<D> &padding() const noexcept#

inline ExpandingArray<D> &padding() noexcept#

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

a parameter that controls the stride of elements in the window

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

inline const ExpandingArray<D> &dilation() const noexcept#

inline ExpandingArray<D> &dilation() noexcept#

inline auto ceil_mode(const bool &new_ceil_mode) -> decltype(*this)#

when True, will use ceil instead of floor to compute the output shape

inline auto ceil_mode(bool &&new_ceil_mode) -> decltype(*this)#

inline const bool &ceil_mode() const noexcept#

inline bool &ceil_mode() noexcept#