code: def log_normalize(a, axis=None):
      
      with np.errstate(under="ignore"):
          a_lse = logsumexp(a, axis)
      a -= a_lse[:, np.newaxis]
docs: Normalizes the input array so that the exponent of the sum is 1.
  
      Parameters
      ----------
      a : array
          Non-normalized input data.
  
      axis : int
          Dimension along which normalization is performed.
  
      Notes
      -----
      Modifies the input **inplace**.