Performs bias correction, dividing by 1-decay^count.
Uses array operations instead of .item() so this function is compatible with JIT tracing (count can be a tracer).
.item()
Performs bias correction, dividing by 1-decay^count.
Uses array operations instead of
.item()so this function is compatible with JIT tracing (count can be a tracer).