We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
jax.nn.one_hot(x, num_classes, dtype)
call_p.multiple_results