addScalarToMatrix function

Tensor<Matrix> addScalarToMatrix(Tensor<Matrix> m, Tensor<Scalar> s)

Implementation

Tensor<Matrix> addScalarToMatrix(Tensor<Matrix> m, Tensor<Scalar> s) {
  int numRows = m.shape[0];
  int numCols = m.shape[1];

  Matrix mMat = m.value;
  double sVal = s.value;

  Matrix outValue = [];
  for (int i = 0; i < numRows; i = i + 1) {
    Vector row = [];
    for (int j = 0; j < numCols; j = j + 1) {
      row.add(mMat[i][j] + sVal);
    }
    outValue.add(row);
  }

  Tensor<Matrix> out = Tensor<Matrix>(outValue);

  out.creator = Node(
    [m, s],
        () {
      double sGradSum = 0.0;
      for (int i = 0; i < numRows; i = i + 1) {
        for (int j = 0; j < numCols; j = j + 1) {
          int idx = i * numCols + j;
          m.grad[idx] = m.grad[idx] + out.grad[idx];
          sGradSum = sGradSum + out.grad[idx];
        }
      }
      s.grad[0] = s.grad[0] + sGradSum;
    },
    opName: 'addScalarToMatrix',
    cost: numRows * numCols,
  );
  return out;
}