TensorFlow.js tf.sqrt()函数

TensorFlow.js中的tf.sqrt()函数用于计算一个张量中每个元素的平方根。

语法

tf.sqrt(x)

参数

• x：一个张量。必填。

返回值

返回一个新的张量，其值为x中每个元素的平方根。

示例

const x = tf.tensor1d([4, 9, 16]);
const result = tf.sqrt(x);

result.print();

输出结果为：

Tensor
  [2, 3]
  2
  3
  4

注意事项

• 该函数只支持浮点数类型张量。
• 对于非正数元素，该函数会返回NaN。