mindspore.ops.Mvlgamma

查看源文件
class mindspore.ops.Mvlgamma(p)[源代码]

逐元素计算 p 维多元对数伽马函数值。

警告

这是一个实验性API,后续可能修改或删除。

更多细节请参考 mindspore.ops.mvlgamma()

参数:
  • p (int) - 进行计算的维度,必须大于等于1。

输入:
  • x (Tensor) - 多元对数伽马函数的输入Tensor,支持数据类型为float32和float64。其shape为 \((N,*)\) ,其中 \(*\) 为任意数量的额外维度。 x 中每个元素的值必须大于 \((p - 1) / 2\)

输出:

Tensor。shape和类型与 x 一致。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> x = Tensor(np.array([[3, 4, 5], [4, 2, 6]]), mindspore.float32)
>>> op = ops.Mvlgamma(p=3)
>>> y = op(x)
>>> print(y)
[[ 2.694925   5.402975   9.140645 ]
 [ 5.402975   1.5963125 13.640454 ]]