Trait custos_math::RandOp

source ·
pub trait RandOp<T, S: Shape = (), D: Device = Self>: Device {
    // Required method
    fn rand(&self, x: &mut Buffer<'_, T, D, S>, lo: T, hi: T);
}

Required Methods§

source

fn rand(&self, x: &mut Buffer<'_, T, D, S>, lo: T, hi: T)

Implementors§

source§

impl<T: Float> RandOp<T, (), CUDA> for CUDA

source§

impl<T: Float> RandOp<T, (), OpenCL> for OpenCL

source§

impl<T: Float, D: MainMemory, S: Shape> RandOp<T, S, D> for CPU