Trait rai_core::FromTensorGrads

pub trait FromTensorGrads {
    // Required method
    fn from_tensor_grads(
        tensors: &[Tensor],
        grads: HashMap<usize, Tensor>
    ) -> Self;
}

Required Methods

fn from_tensor_grads(tensors: &[Tensor], grads: HashMap<usize, Tensor>) -> Self

Object Safety

This trait is not object safe.

Implementations on Foreign Types

impl FromTensorGrads for HashMap<usize, Tensor>

fn from_tensor_grads(tensors: &[Tensor], grads: HashMap<usize, Tensor>) -> Self

impl<const N: usize> FromTensorGrads for [Tensor; N]

fn from_tensor_grads(tensors: &[Tensor], grads: HashMap<usize, Tensor>) -> Self

Implementors