Python PyTorch zeta用法及代码示例

本文简要介绍python语言中 torch.special.zeta 的用法。

用法:

torch.special.zeta(input, other, *, out=None) → Tensor

参数：

• input(Tensor) -对应于 x 的输入张量。

• other(Tensor) -对应于 q 的输入张量。

关键字参数：

out(Tensor,可选的) -输出张量。

逐元素计算 Hurwitz zeta 函数。

\zeta(x, q) = \sum_{k=0}^{\infty} \frac{1}{(k + q)^x}

注意

黎曼zeta函数对应q = 1时的情况

例子：

>>> x = torch.tensor([2., 4.])
>>> torch.special.zeta(x, 1)
tensor([1.6449, 1.0823])
>>> torch.special.zeta(x, torch.tensor([1., 2.]))
tensor([1.6449, 0.0823])
>>> torch.special.zeta(2, torch.tensor([1., 2.]))
tensor([1.6449, 0.6449])