當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

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


計算輸入張量的元素平方根。

用法

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

參數

  • x bfloat16 , half , float32 , float64 , complex64 , complex128 類型的 tf.Tensor
  • name 操作的名稱(可選)。

返回

  • A tf.Tensor具有相同的大小、類型和稀疏性x.

    如果 x 是 SparseTensor ，則返回 SparseTensor(x.indices, tf.math.sqrt(x.values, ...), x.dense_shape)

注意：此操作不支持整數類型。

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 的輸入張量。

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.math.sqrt。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。