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


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


将标量乘以 TensorIndexedSlices 对象。

用法

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

参数

  • scalar 一个 0-D 标量 Tensor 。必须有已知的形状。
  • x 要缩放的TensorIndexedSlices
  • name 操作的名称(可选)。

返回

  • scalar * xx 的类型相同(TensorIndexedSlices)。

抛出

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