sum

1 import grain.testing;
2 import numir;
3 import mir.ndslice;
4 import std.meta;
5 
6 auto hx = [1f, 2f, 3f, 4f].sliced(2, 2).variable(true);
7 auto loss = sum(hx);
8 assert(loss.data[0] == 10f);
9 loss.backward();
10 assert(hx.gradSlice == [1f, 1f, 1f, 1f].sliced(2, 2));