当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。