And could you please have a look at another problem I posted, its about the behaviour of Multinomial with input inf and nan and <0 value in difference torch version.Torch 1.6.0 RuntimeError: probability tensor contains either inf, nan or element < 0, But good with Torch 1.1.0