Trait rai_core::Value

pub trait Value: ValueSpec {
    // Required methods
    fn tensors(&self) -> Self::Tensors;
    fn grad(
        tensors: &Self::Tensors,
        grad_map: &HashMap<usize, Tensor>
    ) -> Self::Gradient;
    fn grad_map(
        tensors: &Self::Tensors,
        grad: Self::Gradient,
        out: &mut HashMap<usize, Tensor>
    );

    // Provided method
    fn to_tensor_vec(&self) -> Vec<Tensor> { ... }
}

Required Methods

fn tensors(&self) -> Self::Tensors

fn grad( tensors: &Self::Tensors, grad_map: &HashMap<usize, Tensor> ) -> Self::Gradient

fn grad_map( tensors: &Self::Tensors, grad: Self::Gradient, out: &mut HashMap<usize, Tensor> )

Provided Methods

fn to_tensor_vec(&self) -> Vec<Tensor>

Object Safety

This trait is not object safe.

Implementors

impl<T> Value for T

where T: ValueSpec + GenericValue<T::Kind, T::Tensors, T::Gradient>,