3 Commits

Author	SHA1	Message	Date
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
Laurent Mazare	1f26042693	Move some shared functions to the nn module. (#221)	2023-07-22 13:25:11 +01:00