Function act_probability 

pub fn act_probability(act: [f32; 2]) -> f64

The act head’s first probability, as torch.softmax gives it.