Trait rai_core::ToTensorGrads

pub trait ToTensorGrads {
    // Required method
    fn to_tensor_grads(self, tensors: &[Tensor]) -> HashMap<usize, Tensor>;
}

Required Methods

fn to_tensor_grads(self, tensors: &[Tensor]) -> HashMap<usize, Tensor>

Implementations on Foreign Types

impl ToTensorGrads for HashMap<usize, Tensor>

fn to_tensor_grads(self, tensors: &[Tensor]) -> HashMap<usize, Tensor>

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

fn to_tensor_grads(self, tensors: &[Tensor]) -> HashMap<usize, Tensor>

Implementors