Trait burn_tensor::backend::ADBackend

pub trait ADBackend: Backend {
    type InnerBackend: Backend<Device = Self::Device>;

    fn backward<const D: usize>(tensor: &Self::TensorPrimitive<D>) -> Gradients;
    fn grad<const D: usize>(
        tensor: &Self::TensorPrimitive<D>,
        grads: &Gradients
    ) -> Option<<<Self as ADBackend>::InnerBackend as Backend>::TensorPrimitive<D>>;
    fn inner<const D: usize>(
        tensor: &Self::TensorPrimitive<D>
    ) -> <Self::InnerBackend as Backend>::TensorPrimitive<D>;
    fn from_inner<const D: usize>(
        tensor: <Self::InnerBackend as Backend>::TensorPrimitive<D>
    ) -> Self::TensorPrimitive<D>;
}

Required Associated Types

type InnerBackend: Backend<Device = Self::Device>

Required Methods

fn backward<const D: usize>(tensor: &Self::TensorPrimitive<D>) -> Gradients

fn grad<const D: usize>(
    tensor: &Self::TensorPrimitive<D>,
    grads: &Gradients
) -> Option<<<Self as ADBackend>::InnerBackend as Backend>::TensorPrimitive<D>>

fn inner<const D: usize>(
    tensor: &Self::TensorPrimitive<D>
) -> <Self::InnerBackend as Backend>::TensorPrimitive<D>

fn from_inner<const D: usize>(
    tensor: <Self::InnerBackend as Backend>::TensorPrimitive<D>
) -> Self::TensorPrimitive<D>

Implementors