random_normal

burn_cubecl::kernel::prng

Function random_normal 

Source 
pub fn random_normal<R: CubeRuntime>(
    shape: Shape,
    device: &R::Device,
    mean: f32,
    std: f32,
    dtype: DType,
) -> CubeTensor<R>
Expand description

Pseudo-random generator with uniform distribution