Trait neuronika::Cat

pub trait Cat<Rhs> {
    type Output;
    fn cat(self, other: Rhs, axis: usize) -> Self::Output;
}

Concatenation.

Associated Types

type Output

The type of the concatenation’s result. See the differentiability arithmetic for more details.

Required methods

fn cat(self, other: Rhs, axis: usize) -> Self::Output

Concatenates variables along the given axis.

Implementors

impl<F1, B1, F2> Cat<Var<F2>> for VarDiff<F1, B1> where
    F1: Data<Dim = B1::Dim> + 'static,
    F2: Data<Dim = F1::Dim> + 'static,
    B1: Gradient + Overwrite + 'static,
    F1::Dim: RemoveAxis,
    B1::Dim: RemoveAxis, 

type Output = VarDiff<Concatenate<F1, F2>, ConcatenateBackwardLeft<B1>>

impl<F1, B1, F2, B2> Cat<VarDiff<F2, B2>> for VarDiff<F1, B1> where
    F1: Data + 'static,
    B1: Gradient + Overwrite + 'static,
    F2: Data<Dim = F1::Dim> + 'static,
    B2: Gradient<Dim = B1::Dim> + Overwrite + 'static,
    F1::Dim: RemoveAxis,
    B1::Dim: RemoveAxis, 

type Output = VarDiff<Concatenate<F1, F2>, ConcatenateBackward<B1, B2>>

impl<F1, F2> Cat<Var<F2>> for Var<F1> where
    F1: Data + 'static,
    F2: Data<Dim = F1::Dim> + 'static,
    F1::Dim: RemoveAxis, 

type Output = Var<Concatenate<F1, F2>>

impl<F1, F2, B2> Cat<VarDiff<F2, B2>> for Var<F1> where
    F1: Data<Dim = B2::Dim> + 'static,
    F2: Data<Dim = F1::Dim> + 'static,
    B2: Gradient + Overwrite + 'static,
    F1::Dim: RemoveAxis,
    B2::Dim: RemoveAxis, 

type Output = VarDiff<Concatenate<F1, F2>, ConcatenateBackwardRight<B2>>