Zeta
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR&950DT系列产品 | √ |
| Atlas A3系列产品 | √ |
| Atlas A2系列产品 | √ |
| Atlas 200I/500 A2推理产品 | √ |
| Atlas推理系列产品 | √ |
| Atlas训练系列产品 | √ |
功能说明
-
算子功能:计算Hurwitz zeta函数,即 ζ(x, q) = Σ(n=0 to ∞) 1/(q+n)^x。
-
计算公式:
z=ζ(x,q)=∑n=0∞1(q+n)xz = \zeta(x, q) = \sum_{n=0}^{\infty} \frac{1}{(q+n)^x}
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x | 输入 | 输入张量,Hurwitz zeta函数的指数参数。 | FLOAT、DOUBLE | ND |
| q | 输入 | 输入张量,Hurwitz zeta函数的偏移参数,必须与x具有相同的数据类型。 | FLOAT、DOUBLE | ND |
| z | 输出 | 输出张量,Hurwitz zeta函数的计算结果,与x具有相同的类型和形状。 | FLOAT、DOUBLE | ND |
约束说明
- 输入x和q的数据类型必须一致。
- 输入x和q的数据大小必须一致(不支持broadcast)。
调用说明
| 调用方式 | 样例代码 | 说明 |
|---|---|---|
| 图模式接口 | test_geir_zeta | 通过算子IR接口方式调用Zeta算子。 |