Function multinomial_sample 

pub fn multinomial_sample(
    probs: &Tensor,
    temperature: f64,
    seed: u64,
) -> Result<Tensor>