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