當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


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


計算多項式的元素值。

用法

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

參數

  • coeffs Tensor 列表表示多項式的係數。
  • x 表示多項式變量的Tensor
  • name 操作的名稱(可選)。

返回

  • 形狀的 tensor 作為表達式 p(x) 應用了元素加法和乘法的常用廣播規則。

如果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。

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.math.polyval。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。