addScalar function

Tensor<Vector> addScalar(
  1. Tensor<Vector> v,
  2. double s
)

Implementation

Tensor<Vector> addScalar(Tensor<Vector> v, double s) {
  int N = v.data.length;
  Vector outValue = [];
  for (int i = 0; i < N; i = i + 1) {
    outValue.add(v.data[i] + s);
  }
  Tensor<Vector> out = Tensor<Vector>(outValue);
  out.creator = Node(
    [v],
        () {
      for (int i = 0; i < N; i = i + 1) {
        v.grad[i] = v.grad[i] + out.grad[i];
      }
    },
    opName: 'addScalar_vector',
    extraParams: {'s': s},
    cost: N,
  );
  return out;
}