sumMatrix function

Tensor<Scalar> sumMatrix(Tensor<Matrix> m)

Implementation

Tensor<Scalar> sumMatrix(Tensor<Matrix> m) {
  int length = m.data.length;
  double total = 0.0;

  for (int i = 0; i < length; i = i + 1) {
    total = total + m.data[i];
  }

  Tensor<Scalar> out = Tensor<Scalar>(total);

  out.creator = Node(
    [m],
        () {
      for (int i = 0; i < length; i = i + 1) {
        m.grad[i] = m.grad[i] + out.grad[0];
      }
    },
    opName: 'sum_matrix',
    cost: length,
  );
  return out;
}