random_sparse_tensor_data_with_rng

Function random_sparse_tensor_data_with_rng 

Source
pub fn random_sparse_tensor_data_with_rng<R>(
    dims: &[usize],
    sparsity: Option<f32>,
    rng: &mut R,
) -> TensorData
where R: Rng,
Expand description

Generates random sparse [DataTensor] object. Fills in sparse tensor based on sparsity value (defaults to 0.5).

ยงExamples

let shape = vec![5, 4, 3];
random_sparse_tensor_data_with_rng(&shape, None, &mut rand::thread_rng());