提交 df7fcb21 authored 作者: Ricardo's avatar Ricardo 提交者: Brandon T. Willard

Add digamma alias to psi Elemwise Op

上级 c80dd0e6
...@@ -1348,6 +1348,9 @@ def psi(a): ...@@ -1348,6 +1348,9 @@ def psi(a):
"""derivative of log gamma function""" """derivative of log gamma function"""
digamma = psi
@scalar_elemwise @scalar_elemwise
def tri_gamma(a): def tri_gamma(a):
"""second derivative of the log gamma function""" """second derivative of the log gamma function"""
...@@ -2897,6 +2900,7 @@ __all__ = [ ...@@ -2897,6 +2900,7 @@ __all__ = [
"gamma", "gamma",
"gammaln", "gammaln",
"psi", "psi",
"digamma",
"tri_gamma", "tri_gamma",
"chi2sf", "chi2sf",
"gammainc", "gammainc",
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论