uniform

shrew::nn::init

Function uniform 

Source 
pub fn uniform<B>(
    shape: impl Into<Shape>,
    low: f64,
    high: f64,
    dtype: DType,
    device: &<B as Backend>::Device,
) -> Result<Tensor<B>, Error>
where
    B: Backend,
Expand description

Initialize a tensor from a uniform distribution U(low, high).