Skip to main content

ConvIO

Trait ConvIO 

Source
pub trait ConvIO {
    type Output;
    type Input;
    type OutputShape;
    type InputShape;
    type FilterShape;

    const N: usize;
}
Expand description

Type-level input/output tensor metadata for conv layers.

Required Associated Constants§

Source

const N: usize

Required Associated Types§

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors§

Source§

impl<const IW: usize, const IH: usize, const IC: usize, const FH: usize, const FW: usize, const OC: usize, const S: usize, const P: usize> ConvIO for Conv<IW, IH, IC, FH, FW, OC, S, P>
where [(); { _ }]:,

Source§

const N: usize

Source§

type Input = Tensor<Dim<IC, Dim<IH, Dim<IW, Nil>>>>

Source§

type Output = Tensor<<Conv<IW, IH, IC, FH, FW, OC, S, P> as ConvIO>::OutputShape>

Source§

type InputShape = Dim<IC, Dim<IH, Dim<IW, Nil>>>

Source§

type OutputShape = Dim<OC, Dim<{ $first }, Dim<{ $dim }, Nil>>>

Source§

type FilterShape = Dim<FH, Dim<FW, Dim<IC, Nil>>>