Skip to content
Discussion options

You must be logged in to vote

The gradient simplifies to Softmax(logits) - one_hot(targets), with zero auxiliary tensor allocations.

Replies: 1 comment

Comment options

Ultron09
Aug 28, 2026
Maintainer Author

You must be logged in to vote
0 replies
Answer selected by Ultron09
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
1 participant