def grad(x): grad = np.zeros_like(x) ldirichlet_softmax(x, alpha, grad=grad) return grad
def func(x): return ldirichlet_softmax(x, alpha)
def logprob(x, alpha): grad = np.zeros_like(x) logp = ldirichlet_softmax(x, alpha, grad=grad) return logp, grad