TensorFlow.js tf.pow()函数

TensorFlow.js 是一个用于在浏览器和 Node.js 中训练和使用机器学习和深度学习模型的 JavaScript 库。其中，tf.pow() 函数用于计算两个张量的元素间的幂运算。

语法

tf.pow(base, exponent)

• base(必选参数)：需要进行幂运算的张量。
• exponent(必选参数)：进行幂运算指数的张量。

返回值

返回一个张量，张量的形状（shape）和输入的张量相同，元素的值为输入张量中的每个元素的 base 次幂。

示例

const base = tf.tensor1d([1, 2, 3, 4]);
const exponent = tf.tensor1d([2, 2, 2, 2]);
const result = tf.pow(base, exponent);
result.print();

输出：

Tensor
    [1, 2, 3, 4]
    dtype: float32
    shape: [4]
    values:
    [ 1.,  4.,  9., 16.]

在上述示例中，我们创建了长度为 4 的两个张量 base 和 exponent，并计算出 base 中的每个元素的 2 次幂，最后将结果保存到 result 张量中并输出。输出结果显示，我们成功地计算出了 base 中每个元素的 2 次幂。

注意事项

• tf.pow() 函数支持 broadcast（广播）机制，即在进行幂运算时，张量的形状可以不同，但必须满足广播规则。