Python - tensorflow.math.igamma()

介绍

tensorflow.math.igamma() 是 TensorFlow 中的一个函数，用于计算自然参数 $a$ 和给定输入 $x$ 的不完全伽玛函数。该函数是 $\Gamma(a, x)/\Gamma(a)$ 的值，其中 $\Gamma(a, x)$ 是伽玛函数的不完全积分形式。

该函数 often 用于正态分布和混合高斯模型的实现中。

语法

tf.math.igamma(a, x, name=None)

参数说明:

• a: 自然参数, 必须是 TensorFlow 的浮点类型, 具有 dtype 属性值。
• x: 输入值，必须是 TensorFlow 的浮点类型, 具有 dtype 属性值, 且必须与 a 具有相同的 dtype 属性值。
• name: 可选参数, 操作名称 (用于 TensorBoard 等可视化工具)。

返回值:

• 一个张量类型的值, 类型与 a 和 x 具有相同的 dtype 属性值。

示例

示例 1

import tensorflow as tf

a = tf.constant(2.0)
x = tf.constant(3.0)
result = tf.math.igamma(a, x)

print(result) # tf.Tensor(0.32332358, shape=(), dtype=float32)

在这个例子中，我们计算了 $\Gamma(2,3)/\Gamma(2)$ 的值，结果为 0.32332358。

示例 2

import tensorflow as tf

a = tf.range(5)[:, tf.newaxis]
x = tf.range(5)[tf.newaxis, :]
result = tf.math.igamma(a, x)

print(result)
'''
tf.Tensor(
[[0.0000000e+00 0.0000000e+00 0.0000000e+00 0.0000000e+00 0.0000000e+00]
 [1.0000000e+00 7.3575879e-01 4.4835938e-01 2.4532535e-01 1.2330032e-01]
 [1.0000000e+00 9.5180891e-01 8.6896776e-01 7.2583724e-01 5.4213421e-01]
 [1.0000000e+00 9.8201374e-01 9.3224835e-01 8.4773673e-01 7.1637459e-01]
 [1.0000000e+00 9.9303195e-01 9.6836826e-01 9.1892225e-01 8.4551691e-01]], shape=(5, 5), dtype=float32)
'''

在这个例子中，我们计算了自然参数为 $[0,1,2,3,4]$，输入值为 $[0,1,2,3,4]$ 的不完全伽马函数的值。函数返回一个张量，每行对应一个自然参数，每列对应一个输入值。

创建者

这个函数首先出现在 TensorFlow v1.14 中，并由 tensorflow.math.igamma() 实现。

参考文献

• TensorFlow 文档，tf.math.igamma。