akichinguyen issues

Repositories
Issues
Comments

Results 1 issues of


                                            akichinguyen

Torch.nn.functional.one_hot does not work as expected with ROCM

### 🐛 Describe the bug I installed the pytorch 2.1 with rocm 5.4.2. When I tested the one_hot function, it returned this: print(torch.nn.functional.one_hot(torch.tensor(0,device=‘cuda:0’), 2)) = [0,0] Do you know why...