Python – tensorflow.math.rsqrt()

0 0 1 minute read

TensorFlow is open-source python library designed by Google to develop Machine Learning models and deep learning neural networks. rsqrt() is used to find element-wise reciprocal of square root of x.

Syntax: tf.math.rsqrt(x, name)

Parameter:

x: It’s the input tensor. Allowed dtype for this tensor are bfloat16, half, float32, float64.

name(optional): It defines the name for the operation.

Returns: It returns a tensor of same dtype as x.

Example 1:

Python3

# Importing the library
import tensorflow as tf
 
# Initializing the input tensor
a = tf.constant([.2, 1.6, 3.7, 1], dtype = tf.float64)
 
# Printing the input tensor
print('a: ', a)
 
# Calculating result
res = tf.math.rsqrt(a)
 
# Printing the result
print('Result: ', res)

Output:

a:  tf.Tensor([0.2 1.6 3.7 1. ], shape=(4, ), dtype=float64)
Result:  tf.Tensor([2.23606798 0.79056942 0.51987524 1.        ], shape=(4, ), dtype=float64)

Example 2:

Python3

# importing the library
import tensorflow as tf
 
# Initializing the input tensor
a = tf.constant([2.5, 5.5, 1.5, 6.5], dtype = tf.float64)
 
# Printing the input tensor
print('a: ', a)
 
# Calculating result
res = tf.math.rsqrt(a)
 
# Printing the result
print('Result: ', res)

Output:

a:  tf.Tensor([2.5 5.5 1.5 6.5], shape=(4, ), dtype=float64)
Result:  tf.Tensor([0.63245553 0.42640143 0.81649658 0.39223227], shape=(4, ), dtype=float64)