tensorflow_probability/python/distributions/wishart.py
def _multi_digamma(self, a, p, name="multi_digamma"):
    """Computes the multivariate digamma function; Psi_p(a)."""
    with self._name_scope(name):
      seq = self._multi_gamma_sequence(a, p)
      return tf.reduce_sum(input_tensor=tf.math.digamma(seq), axis=[-1])
def _multi_digamma(self, a, p, name="multi_digamma"):
    """Computes the multivariate digamma function; Psi_p(a)."""
    with self._name_scope(name):
      seq = self._multi_gamma_sequence(a, p)
      return tf.reduce_sum(input_tensor=tf.math.digamma(seq), axis=[-1])
def _multi_digamma ( self , a , p , name = "multi_digamma" ) : with self . _name_scope ( name ) : seq = self . _multi_gamma_sequence ( a , p ) return tf . reduce_sum ( input_tensor = tf . math . digamma ( seq ) , axis = [ - 1 ] )
Computes the multivariate digamma function; Psi_p(a).
Computes the multivariate digamma function ; Psi_p ( a ) .