zeroGraphGrads method

void zeroGraphGrads(
  1. CommandBuffer tape
)

Implementation

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