Struct burn_tensor::ops::ConvOptions

pub struct ConvOptions<const N: usize> {
    pub stride: [usize; N],
    pub padding: [usize; N],
    pub dilation: [usize; N],
    pub groups: usize,
}

Convolution options.

Fields

stride: [usize; N]

Stride.

padding: [usize; N]

Padding.

dilation: [usize; N]

Dilation.

groups: usize

Groups.

Implementations

impl<const N: usize> ConvOptions<N>

pub fn new( stride: [usize; N], padding: [usize; N], dilation: [usize; N], groups: usize, ) -> Self

Constructs a new ConvOptions.

Trait Implementations

impl<const N: usize> Clone for ConvOptions<N>

fn clone(&self) -> ConvOptions<N>

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<const N: usize> Debug for ConvOptions<N>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl From<Conv1dOptionsDescription> for ConvOptions<1>

fn from(val: Conv1dOptionsDescription) -> Self

Converts to this type from the input type.

impl From<Conv2dOptionsDescription> for ConvOptions<2>

fn from(val: Conv2dOptionsDescription) -> Self

Converts to this type from the input type.

impl From<Conv3dOptionsDescription> for ConvOptions<3>

fn from(val: Conv3dOptionsDescription) -> Self

Converts to this type from the input type.

impl From<ConvOptions<1>> for Conv1dOptionsDescription

fn from(value: ConvOptions<1>) -> Self

Converts to this type from the input type.

impl From<ConvOptions<2>> for Conv2dOptionsDescription

fn from(value: ConvOptions<2>) -> Self

Converts to this type from the input type.

impl From<ConvOptions<3>> for Conv3dOptionsDescription

fn from(value: ConvOptions<3>) -> Self

Converts to this type from the input type.

impl<const N: usize> Hash for ConvOptions<N>

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl<const N: usize> PartialEq for ConvOptions<N>

fn eq(&self, other: &ConvOptions<N>) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl<const N: usize> Eq for ConvOptions<N>

impl<const N: usize> StructuralPartialEq for ConvOptions<N>