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
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
