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