jax.nn.one_hot: deprecate non-integer inputs#25590

Merged

copybara-service[bot] merged 1 commit intojax-ml:mainfrom jakevdp:fix-one-hot-float

Dec 19, 2024

+34-15

Commits on Dec 19, 2024

jax.nn.one_hot: deprecate non-integer inputs
jakevdp
committed