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

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

將標量乘以 Tensor 或 IndexedSlices 對象。

用法

tf.math.scalar_mul(
    scalar, x, name=None
)

參數

  • scalar 一個 0-D 標量 Tensor 。必須有已知的形狀。
  • x 要縮放的Tensor 或IndexedSlices。
  • name 操作的名稱(可選)。

返回

  • scalar * x 與 x 的類型相同(Tensor 或 IndexedSlices)。

拋出

  • ValueError 如果標量不是 0-D scalar 。

這是 tf.math.multiply 的特殊情況，其中第一個值必須是 scalar 。與 tf.math.multiply 的一般形式不同，這是保證對 tf.IndexedSlices 有效的操作。

x = tf.reshape(tf.range(30, dtype=tf.float32), [10, 3])
with tf.GradientTape() as g:
  g.watch(x)
  y = tf.gather(x, [1, 2])  # IndexedSlices
  z = tf.math.scalar_mul(10.0, y)

相關用法


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