Function sum
Computes a sum reduction along the specified axes.
Parameters
Name | Description |
---|---|
op | The input to the reduction. |
axes | The axes the reduction should be performed along. |
Returns
The resulting operation.
Example
import dopt .core : evaluate;
auto s1 = float32([2], [0.5, 1.5]) .sum();
auto s2 = float32([2, 2], [0, 1, 0, 5]) .sum();
auto s3 = float32([2, 2], [0, 1, 0, 5]) .sum([0]);
auto s4 = float32([2, 2], [0, 1, 0, 5]) .sum([1]);
assert(s1 .evaluate() .as!float == [2.0f]);
assert(s2 .evaluate() .as!float == [6.0f]);
assert(s3 .evaluate() .as!float == [0.0f, 6.0f]);
assert(s4 .evaluate() .as!float == [1.0f, 5.0f]);