def METHOD_NAME(self):
    """Reset all of the metric state variables.
    This function is called between epochs/steps,
    when a metric is evaluated during training.
    """
    for v in self.variables:
        v.assign(ops.zeros(v.shape, dtype=v.dtype))