zeroGraphGrads method

void zeroGraphGrads(
  1. CommandBuffer tape
)

Finds every GPUTensor the current GPUTensor depends on and sets their gradients to 0.

Implementation

void zeroGraphGrads(CommandBuffer tape) {
  List<GPUTensor> allTensors = getAllTensorsInGraph(this);
  for (int i = 0; i < allTensors.length; i = i + 1) {
    allTensors[i].zeroGrad(tape);
  }
}