pub struct MaxPoolOptions<const N: usize> {
pub kernel_size: [usize; N],
pub stride: [usize; N],
pub padding: [(usize, usize); N],
pub dilation: [usize; N],
pub ceil_mode: bool,
}Expand description
Max pooling options.
Fields§
§kernel_size: [usize; N]Kernel size (non-zero).
stride: [usize; N]Stride (non-zero).
padding: [(usize, usize); N]Padding as (begin, end) pairs for each spatial dimension.
dilation: [usize; N]Dilation (non-zero).
ceil_mode: boolIf true, use ceiling instead of floor for output size calculation.
Implementations§
Source§impl<const N: usize> MaxPoolOptions<N>
impl<const N: usize> MaxPoolOptions<N>
Sourcepub fn new(kernel_size: [usize; N]) -> Self
pub fn new(kernel_size: [usize; N]) -> Self
Constructs max pooling options with the given kernel size.
Defaults to a stride equal to the kernel size, no padding, a dilation of 1 and floor mode.
Sourcepub fn with_stride(self, stride: [usize; N]) -> Self
pub fn with_stride(self, stride: [usize; N]) -> Self
Sets the stride.
Sourcepub fn with_padding(self, padding: [usize; N]) -> Self
pub fn with_padding(self, padding: [usize; N]) -> Self
Sets symmetric padding for every spatial dimension.
Sourcepub fn with_padding_pairs(self, padding: [(usize, usize); N]) -> Self
pub fn with_padding_pairs(self, padding: [(usize, usize); N]) -> Self
Sets explicit (begin, end) padding for every spatial dimension.
Sourcepub fn with_dilation(self, dilation: [usize; N]) -> Self
pub fn with_dilation(self, dilation: [usize; N]) -> Self
Sets the dilation.
Sourcepub fn with_ceil_mode(self, ceil_mode: bool) -> Self
pub fn with_ceil_mode(self, ceil_mode: bool) -> Self
Sets whether to use ceiling instead of floor for output size calculation.
Sourcepub fn is_asymmetric(&self) -> bool
pub fn is_asymmetric(&self) -> bool
Returns true if padding is asymmetric.
Trait Implementations§
Source§impl<const N: usize> Clone for MaxPoolOptions<N>
impl<const N: usize> Clone for MaxPoolOptions<N>
Source§impl<const N: usize> Debug for MaxPoolOptions<N>
impl<const N: usize> Debug for MaxPoolOptions<N>
impl<const N: usize> Eq for MaxPoolOptions<N>
Source§impl<const N: usize> Hash for MaxPoolOptions<N>
impl<const N: usize> Hash for MaxPoolOptions<N>
Source§impl<const N: usize> PartialEq for MaxPoolOptions<N>
impl<const N: usize> PartialEq for MaxPoolOptions<N>
impl<const N: usize> StructuralPartialEq for MaxPoolOptions<N>
Auto Trait Implementations§
impl<const N: usize> Freeze for MaxPoolOptions<N>
impl<const N: usize> RefUnwindSafe for MaxPoolOptions<N>
impl<const N: usize> Send for MaxPoolOptions<N>
impl<const N: usize> Sync for MaxPoolOptions<N>
impl<const N: usize> Unpin for MaxPoolOptions<N>
impl<const N: usize> UnsafeUnpin for MaxPoolOptions<N>
impl<const N: usize> UnwindSafe for MaxPoolOptions<N>
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<Q, K> Equivalent<K> for Q
impl<Q, K> Equivalent<K> for Q
Source§fn equivalent(&self, key: &K) -> bool
fn equivalent(&self, key: &K) -> bool
Compare self to
key and return true if they are equal.