tf.math.reciprocal_no_nan

View source on GitHub

Performs a safe reciprocal operation, element wise.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.math.reciprocal_no_nan, `tf.compat.v2.math.reciprocal_no_nan`

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

If a particular element is zero, the reciprocal for that element is also set to zero.

For example:

x = tf.constant([2.0, 0.5, 0, 1], dtype=tf.float32)
tf.math.reciprocal_no_nan(x)  # [ 0.5, 2, 0.0, 1.0 ]

Args
x	A Tensor of type float16, float32, float64 complex64 or complex128.
name	A name for the operation (optional).

Returns
A Tensor of same shape and type as x.

Raises
TypeError	x must be of a valid dtype.