当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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


按元素计算 x 平方根的倒数。

用法

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

参数

  • x 一个tf.Tensor。必须是以下类型之一:bfloat16 , half , float32 , float64
  • name 操作的名称(可选)。

返回

  • 一个tf.Tensor。具有与 x 相同的类型。

例如:

x = tf.constant([2., 0., -2.])
tf.math.rsqrt(x)
<tf.Tensor:shape=(3,), dtype=float32,
numpy=array([0.707, inf, nan], dtype=float32)>

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.math.rsqrt。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。