Python - tensorflow.math.scalar_mul()

简介

tensorflow.math.scalar_mul() 是 tensorflow 的一个数学函数，用于将一个张量（tensor）乘以一个标量。

语法

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

其中， scalar 是一个标量；x 是一个张量。

返回值

返回值为一个张量，其每个元素都是 x 对应元素乘以标量 scalar 的结果。

示例

import tensorflow as tf

# 定义一个 2x3 的张量
x = tf.constant([[1, 2, 3], [4, 5, 6]], dtype=tf.float32)

# 将张量乘以标量 2
y = tf.math.scalar_mul(2, x)

print(y)

# 输出结果为：
# tf.Tensor(
# [[ 2.  4.  6.]
#  [ 8. 10. 12.]], shape=(2, 3), dtype=float32)

备注

tf.math.scalar_mul() 函数还可以通过设置 name 参数来指定操作的名称。