torch.multinomial(prob, n, replacement)
当 replacement = False 时为非放回采样,此时 n 如果大于 prob 每行非零元素数的时候会报错。
当 replacement = True 时为放回采样,此时 n 大于 prob 每行非零元素数的时候不会报错。但是如果 prob 有某些行为全0则会报错。
torch.multinomial(prob, n, replacement)
当 replacement = False 时为非放回采样,此时 n 如果大于 prob 每行非零元素数的时候会报错。
当 replacement = True 时为放回采样,此时 n 大于 prob 每行非零元素数的时候不会报错。但是如果 prob 有某些行为全0则会报错。