Python tf.math.polyval用法及代码示例

计算多项式的元素值。

用法

tf.math.polyval(
    coeffs, x, name=None
)

如果x 是张量并且coeffs 是列表n + 1 张量，则此函数返回n-th 阶多项式的值

p(x) = coeffs[n-1] + coeffs[n-2] * x + ... + coeffs[0] * x**(n-1)

使用霍纳的方法进行评估，即

p(x) = coeffs[n-1] + x * (coeffs[n-2] + ... + x * (coeffs[1] + x * coeffs[0]))

使用示例：

coefficients = [1.0, 2.5, -4.2]
x = 5.0
y = tf.math.polyval(coefficients, x)
y
<tf.Tensor:shape=(), dtype=float32, numpy=33.3>

使用示例：

tf.math.polyval([2, 1, 0], 3) # evaluates 2 * (3**2) + 1 * (3**1) + 0 * (3**0)
<tf.Tensor:shape=(), dtype=int32, numpy=21>

tf.math.polyval 也可用于多项式回归。与显式写出相比，利用此函数可以更方便地编写多项式方程，尤其是对于更高次多项式。

x = tf.constant(3)
theta1 = tf.Variable(2)
theta2 = tf.Variable(1)
theta3 = tf.Variable(0)
tf.math.polyval([theta1, theta2, theta3], x)
<tf.Tensor:shape=(), dtype=int32, numpy=21>

numpy 兼容性

相当于 numpy.polyval。