# importing the library
import tensorflow as tf
import numpy as np
 
# Initializing the input tensor
a = tf.constant([7, 8, 13, 11, np.inf], dtype = tf.float64)
 
# Printing the input tensor
print('a: ', a)
 
# Calculating the result
res = tf.math.is_inf(a)
 
# Printing the result
print('Result: ', res)

a:  tf.Tensor([ 7.  8. 13. 11. inf], shape=(5, ), dtype=float64)
Result:  tf.Tensor([False False False False  True], shape=(5, ), dtype=bool)

# Importing the library
import tensorflow as tf
import numpy as np
 
# Initializing the input tensor
a = tf.constant([7, 8, 13, 11, np.nan], dtype = tf.float64)
 
# Printing the input tensor
print('a: ', a)
 
# Calculating the result
res = tf.math.is_inf(a)
 
# Printing the result
print('Result: ', res)

a:  tf.Tensor([ 7.  8. 13. 11. nan], shape=(5, ), dtype=float64)
Result:  tf.Tensor([False False False False False], shape=(5, ), dtype=bool)