Struct ConstantPad3d 

pub struct ConstantPad3d<D: Dtype, IT, OT, const RANK: usize> {
    pub pad_d1: usize,
    pub pad_d2: usize,
    pub pad_h1: usize,
    pub pad_h2: usize,
    pub pad_w1: usize,
    pub pad_w2: usize,
    pub value: f64,
    /* private fields */
}

3-D constant padding: pads with a fixed value on each side.

Fields

pad_d1: usize

Padding/config parameter; see the struct docs.

pad_d2: usize

Padding/config parameter; see the struct docs.

pad_h1: usize

Padding/config parameter; see the struct docs.

pad_h2: usize

Padding/config parameter; see the struct docs.

pad_w1: usize

Padding/config parameter; see the struct docs.

pad_w2: usize

Padding/config parameter; see the struct docs.

value: f64

Padding/config parameter; see the struct docs.

Implementations

impl<D: Dtype, IT, OT, const RANK: usize> ConstantPad3d<D, IT, OT, RANK>

pub fn new( pad_d1: usize, pad_d2: usize, pad_h1: usize, pad_h2: usize, pad_w1: usize, pad_w2: usize, value: f64, ) -> Self

Creates a new layer with the given padding/config parameters.

Trait Implementations

impl<D: Dtype, IT: Tensor<D, RANK> + EagerTensor, OT: Tensor<D, RANK>, const RANK: usize> Layer<IT> for ConstantPad3d<D, IT, OT, RANK>

type Output = OT

This layer’s output type.

fn call(&self, _input: IT) -> Self::Output

Applies this layer to input.

impl<D: Dtype, const RANK: usize> Layer<SymTensor> for ConstantPad3d<D, SymTensor, SymTensor, RANK>

type Output = SymTensor

This layer’s output type.

fn call(&self, input: SymTensor) -> SymTensor

Applies this layer to input.