Closed michaelpoluektov closed 4 months ago
Add batched softmax, fix off-by-1 errors with regular softmax when the number of channels is either 2 or 3
Add batched softmax, fix off-by-1 errors with regular softmax when the number of channels is either 2 or 3