def zero_grad

bendtorch-core@0.1.0.0 · src/autograd/tensor_backward.bend:931

import Base

931# Empties every gradient slot, like PyTorch's zero_grad(set_to_none=True).
932def zero_grad(tape: Tape) -> Tape:
933  Tape{entries} = tape
934  Tape{cleared(entries)}