zeroGrad method

void zeroGrad()

Traverses the computational Graph which created this Tensor, setting every value of every gradient to 0.0. This method in machine learning is to be used in between training iterations. Forgetting this method may result in oscillating or exploding gradient values.

Implementation

void zeroGrad() {
  for (int i = 0; i < grad.length; i = i + 1) {
    grad[i] = 0.0;
  }
}