Python - tensorflow.math.polygamma()

介绍

tensorflow.math.polygamma(n, x, name=None) 是TensorFlow中的一个函数，用于计算polygamma函数的值。

Polygamma函数是指数学函数的一种，通常用于统计学和概率论领域。它们是Riemann zeta函数的导数，并在计算分布函数、特征函数、熵和其他统计量时发挥重要作用。

语法

tensorflow.math.polygamma(n, x, name=None)

参数

• n：一个表示多次迭代的正整数张量。
• x：一个表示计算polygamma函数值的实数张量。
• name：可选参数，String类型表示操作的名称。

返回值

• 一个表示polygamma函数值的张量。

示例

import tensorflow as tf

n = tf.constant(2, dtype=tf.float32)
x = tf.constant(1.5)

result = tf.math.polygamma(n, x)

print(result)

输出：

tf.Tensor(3.1281826, shape=(), dtype=float32)

注意事项

• 参数 n 和 x 必须具有相同的数据类型。
• n 必须是一个正整数。
• 使用该函数时请确保输入张量的形状符合您的期望。
• 该函数支持自动微分机制，可以用于构建梯度下降模型进行训练。

参考资料

• TensorFlow官方文档：tensorflow.math.polygamma