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

计算输入张量的元素平方根。

用法

tf.math.sqrt(
    x, name=None
)

注意：此操作不支持整数类型。

x = tf.constant([[4.0], [16.0]])
tf.sqrt(x)
<tf.Tensor:shape=(2, 1), dtype=float32, numpy=
  array([[2.],
         [4.]], dtype=float32)>
y = tf.constant([[-4.0], [16.0]])
tf.sqrt(y)
<tf.Tensor:shape=(2, 1), dtype=float32, numpy=
  array([[nan],
         [ 4.]], dtype=float32)>
z = tf.constant([[-1.0], [16.0]], dtype=tf.complex128)
tf.sqrt(z)
<tf.Tensor:shape=(2, 1), dtype=complex128, numpy=
  array([[0.0+1.j],
         [4.0+0.j]])>

注意：为了支持复杂类型，请提供 complex64 或 complex128 的输入张量。