Junming Chen 569d114ef7
Fix device problem
Before, the one_hot could only run in device='cuda'. Now it will run on input device automatically.
2022-04-19 11:53:18 +08:00
..
2022-04-19 11:53:18 +08:00