Python – tensorflow.math.digamma()

TensorFlow是谷歌设计的开源Python库，用于开发机器学习模型和深度学习神经网络。

digamma()用于计算Lgamma的元素导数，即Gamma(x)绝对值的对数。

语法: tensorflow.math.digamma( x, name)

参数:

• x:是输入的张量。允许的dtypes是bfloat16, half, float32, float64。
• name（可选）:它定义了该操作的名称。

返回：它返回一个与x具有相同dtype的张量。

示例 1:

# importing the library
import tensorflow as tf
  
# Initializing the input tensor
a = tf.constant([1, 2, 3, 4, 5], dtype = tf.float64)
  
# Printing the input tensor
print('Input: ', a)
  
# Calculating digamma
res = tf.math.digamma(x = a)
  
# Printing the result
print('Result: ', res)

输出:

Input:  tf.Tensor([1. 2. 3. 4. 5.], shape=(5, ), dtype=float64)
Result:  tf.Tensor([-0.57721566  0.42278434  0.92278434  1.25611767  1.50611767], shape=(5, ), dtype=float64)

例子2：视觉化

# importing the library
import tensorflow as tf
import matplotlib.pyplot as plt
  
# Initializing the input tensor
a = tf.constant([1, 2, 3, 4, 5], dtype = tf.float64)
  
# Calculating digamma
res = tf.math.digamma(x = a)
  
# Plotting the graph
plt.plot(a, res, color ='green')
plt.title('tensorflow.math.digamma')
plt.xlabel('Input')
plt.ylabel('Result')
plt.show()

输出: