add function

Tensor<Scalar> add(
  1. Tensor<Scalar> a,
  2. Tensor<Scalar> b
)

Implementation

Tensor<Scalar> add(Tensor<Scalar> a, Tensor<Scalar> b) {
  Scalar outValue = a.value + b.value;
  Tensor<Scalar> out = Tensor<Scalar>(outValue);
  out.creator = Node([a, b], () {
    /* backwardFn */
    a.grad += out.grad;
    b.grad += out.grad;
  }, opName: 'add');
  return out;
}