zeroGrad method

void zeroGrad()

Implementation

void zeroGrad() {
  final visited = <Value>{};
  // ignore: no_leading_underscores_for_local_identifiers
  void _reset(Value v) {
    if (!visited.contains(v)) {
      visited.add(v);
      v.grad = 0.0;
      for (final child in v._prev) {
        _reset(child);
      }
    }
  }

  _reset(this);
}