pub fn rand_tensor(shape: &[usize], low: f32, high: f32) -> Result<Tensor>
Generate random tensor with uniform distribution