code: def _kumaraswamy_moment(a, b, n):
      a = np.asarray(a)
      b = np.asarray(b)
      return b * sp_special.beta(1.0 + n / a, b)
  
imports: import numpy
  from scipy import special
  from scipy import stats
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import kumaraswamy
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 2 for calling a global function with example usage: _kumaraswamy_moment(a, b, n) and returns: unknown