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

返回數字符號的元素指示。

用法

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

參數

x 張量。必須是以下類型之一：bfloat16、half、float32、float64、int32、int64、complex64、complex128。
name 操作的名稱(可選)。

張量。與 x 具有相同的類型。
如果 x 是 SparseTensor，則返回 SparseTensor(x.indices, tf.math.sign(x.values, ...), x.dense_shape)。

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

y = sign(x) = -1 if x < 0; 0 if x == 0; 1 if x > 0 。

對於複數，y = sign(x) = x / |x| if x != 0, otherwise y = 0。

示例用法：

# real number
tf.math.sign([0., 2., -3.])
<tf.Tensor:shape=(3,), dtype=float32,
numpy=array([ 0.,  1., -1.], dtype=float32)>

# complex number
tf.math.sign([1 + 1j, 0 + 0j])
<tf.Tensor:shape=(2,), dtype=complex128,
numpy=array([0.70710678+0.70710678j, 0.        +0.j        ])>

相關用法

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