There is support for half in torch.multinomial. And whether torch.distributions.multinomial supports Half(FP16) on GPU and CPU?
There is support for half in torch.multinomial. And whether torch.distributions.multinomial supports Half(FP16) on GPU and CPU?