This repository was archived by the owner on Nov 17, 2025. It is now read-only.
This repository was archived by the owner on Nov 17, 2025. It is now read-only.
Move Softmax and related Ops out of nnet module #764
Description
This was meant to be done after #673
We should move Softmax/ LogSoftmax and SoftmaxGrad (and their rewrites) out of the nnet submodule and into the tensor module. We should also create an alias for logsoftmax -> log_softmax to match the scipy name.