AtomicVar 755dcf6137 Enable cross_entropy loss to handle dense targets (#517) ...

* Enable cross_entropy loss to handle dense targets

Dense targets means probabilities or one-hot encodings.

* better shape check of weights

* nits in docstring

---------

Co-authored-by: Awni Hannun <awni@apple.com>

2024-01-23 12:17:22 -08:00

..

mlx

Enable cross_entropy loss to handle dense targets (#517)

2024-01-23 12:17:22 -08:00

src

use axes in tensordot (#525)

2024-01-22 21:17:00 -08:00

tests

Enable cross_entropy loss to handle dense targets (#517)

2024-01-23 12:17:22 -08:00