Trait rai_nn::init::Init

pub trait Init {
    // Required method
    fn new_tensor<T: Type>(
        &self,
        shape: impl Shape,
        dtype: T,
        device: impl AsDevice
    ) -> Tensor;
}

Required Methods

fn new_tensor<T: Type>( &self, shape: impl Shape, dtype: T, device: impl AsDevice ) -> Tensor

Object Safety

This trait is not object safe.

Implementors

impl Init for Kaiming

impl Init for Uniform