Hello
How are you?
Thanks for contributing to this project.
I found a bug in module for GPU.

The script that I ran is the following:

import paddle
import paddle.nn.functional as F

a = paddle.randint(0, 10, (4, 4))
b = F.one_hot(a, 5)
print(a)
print(b)

The above script does NOT yield any issue when running on GPU.
But when running on CPU it yields an issue.