torch:sum
(torch:sum a &key axis keepdims)
Differentiable sum, of every element (no :axis) or along an axis, following linalg:sum's :axis / :keepdims rules. The backward pass broadcasts the gradient back over the reduced extent.
(torch:sum a &key axis keepdims)
Differentiable sum, of every element (no :axis) or along an axis, following linalg:sum's :axis / :keepdims rules. The backward pass broadcasts the gradient back over the reduced extent.