def zero_grad

bendtorch-core@0.1.0.0 · src/autograd/backward.bend:253

import Base

253# Empties every gradient slot, like PyTorch's zero_grad(set_to_none=True).
254def zero_grad(tape: Tape) -> Tape:
255  Tape{entries} = tape
256  Tape{cleared(entries)}