3 Commits

Author	SHA1	Message	Date
Laurent Mazare	eab54e4490	Fix the tests for mkl. (#437)	2023-08-14 08:09:27 +01:00
Laurent Mazare	1064b9b031	Add the cross-entropy loss. (#287)	2023-07-31 14:26:36 +01:00
Laurent Mazare	ffeafbfc43	Make the nll op closer to the pytorch version + add a test. (#286)	2023-07-31 14:14:01 +01:00