code: def _write_avg_norm(self, norms:[])->None:
          "Writes the average norm of the gradients to Tensorboard."
          avg_norm = sum(norms)/len(self.gradients)
          self._add_gradient_scalar('avg_norm', scalar_value=avg_normf _write_avg_norm(self, norms:[])->None:
          "Writes the average norm of the gradients to Tensorboard."
          avg_norm = sum(norms)/len(self.gradients)
          self._add_gradient_scalar('avg_norm', scalar_value=avg_norm)
docs: Writes the average norm of the gradients to Tensorboard.