DeviceStorageCreation

Trait DeviceStorageCreation 

Source
pub trait DeviceStorageCreation: DeviceStorage {
    // Required methods
    fn zeros(
        shape: Vec<usize>,
        device: Option<Device>,
        requires_grad: Option<bool>,
    ) -> Self;
    fn ones(
        shape: Vec<usize>,
        device: Option<Device>,
        requires_grad: Option<bool>,
    ) -> Self;
    fn from_ndarray<S, D, T>(
        data: &ArrayBase<S, D>,
        device: Option<Device>,
        requires_grad: Option<bool>,
    ) -> Self
       where S: Data<Elem = T>,
             T: AsPrimitive<f32>,
             D: Dimension;
    fn uniform(
        l_bound: f32,
        r_bound: f32,
        shape: Vec<usize>,
        device: Option<Device>,
        requires_grad: Option<bool>,
    ) -> Self;
}

Required Methods§

Source

fn zeros( shape: Vec<usize>, device: Option<Device>, requires_grad: Option<bool>, ) -> Self

Source

fn ones( shape: Vec<usize>, device: Option<Device>, requires_grad: Option<bool>, ) -> Self

Source

fn from_ndarray<S, D, T>( data: &ArrayBase<S, D>, device: Option<Device>, requires_grad: Option<bool>, ) -> Self
where S: Data<Elem = T>, T: AsPrimitive<f32>, D: Dimension,

Source

fn uniform( l_bound: f32, r_bound: f32, shape: Vec<usize>, device: Option<Device>, requires_grad: Option<bool>, ) -> Self

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors§