Trait dfdx::tensor::CanPutTapeHolder

source · [−]

pub trait CanPutTapeHolder<H: TapeHolder> {
    type Output;

    fn with_tape_holder(self, tape_holder: H) -> Self::Output;
}

Required Associated Types

type Output

Required Methods

fn with_tape_holder(self, tape_holder: H) -> Self::Output

Implementors

impl<HIn, HOut> CanPutTapeHolder<HOut> for Tensor0D<HIn> where
HIn: TapeHolder,
HOut: TapeHolder,

type Output = Tensor0D<HOut>

impl<const M: usize, HIn, HOut> CanPutTapeHolder<HOut> for Tensor1D<M, HIn> where
HIn: TapeHolder,
HOut: TapeHolder,

type Output = Tensor1D<M, HOut>

impl<const M: usize, const N: usize, HIn, HOut> CanPutTapeHolder<HOut> for Tensor2D<M, N, HIn> where
HIn: TapeHolder,
HOut: TapeHolder,

type Output = Tensor2D<M, N, HOut>

impl<const M: usize, const N: usize, const O: usize, HIn, HOut> CanPutTapeHolder<HOut> for Tensor3D<M, N, O, HIn> where
HIn: TapeHolder,
HOut: TapeHolder,

type Output = Tensor3D<M, N, O, HOut>

impl<const M: usize, const N: usize, const O: usize, const P: usize, HIn, HOut> CanPutTapeHolder<HOut> for Tensor4D<M, N, O, P, HIn> where
HIn: TapeHolder,
HOut: TapeHolder,

type Output = Tensor4D<M, N, O, P, HOut>