Gabe Caldwell bdd6325f31 default num_classes value for one_hot (#6182) ...

* num_classes=-1

If num_classes set to -1, the number of classes will be inferred as one greater than the largest class value in the input tensor.

* num_classes desc

comment to explain num_classes default and what that means.

* replacing ' with `

2024-08-19 12:07:14 -07:00

111 KiB

Raw Blame History

View Raw