Python – tensorflow.math.square()
Last Updated :
14 Mar, 2023
TensorFlow is open-source Python library designed by Google to develop Machine Learning models and deep learning neural networks.
square() is used to compute element wise square of x i.e x*x.
Syntax: tensorflow.math.square(x, name)
Parameters:
- x: It’s a tensor. Allowed dtypes are bfloat16, half, float32, float64, complex64, complex128.
- name(optional): It defines the name for the operation.
Returns: It returns a tensor.
Example 1:
Python3
import tensorflow as tf
a = tf.constant([ - 5 , - 7 , 2 , 5 , 7 ], dtype = tf.float64)
print ( 'a: ' , a)
res = tf.math.square(a)
print ( 'Result: ' , res)
|
Output:
a: tf.Tensor([-5. -7. 2. 5. 7.], shape=(5, ), dtype=float64)
Result: tf.Tensor([25. 49. 4. 25. 49.], shape=(5, ), dtype=float64)
Example 2: Visualization
Python3
import matplotlib.pyplot as plt
a = tf.constant([ - 7 , - 5 , 2 , 5 , 7 ], dtype = tf.float64)
res = tf.math.square(a)
plt.plot(a, res, color = 'green' )
plt.title( 'tensorflow.math.square' )
plt.xlabel( 'Input' )
plt.ylabel( 'Result' )
plt.show()
|
Output:
Share your thoughts in the comments
Please Login to comment...