ferritin_ligandmpnn::ligandmpnn::model

Function multinomial_sample

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