Python – tensorflow.math.square()

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

filter_none

edit
close

play_arrow

link
brightness_4
code

# importing the library
import tensorflow as tf
  
# Initializing the input tensor
a = tf.constant([ -5, -7, 2, 5, 7], dtype = tf.float64)
  
# Printing the input tensor
print('a: ', a)
  
# Calculating result
res = tf.math.square(a)
  
# Printing the result
print('Result: ', res)

chevron_right


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

filter_none

edit
close

play_arrow

link
brightness_4
code

# import tensorflow as tf
import matplotlib.pyplot as plt
  
# Initializing the input tensor
a = tf.constant([ -7, -5, 2, 5, 7], dtype = tf.float64)
  
# Calculating tangent
res = tf.math.square(a)
  
# Plotting the graph
plt.plot(a, res, color ='green')
plt.title('tensorflow.math.square')
plt.xlabel('Input')
plt.ylabel('Result')
plt.show()

chevron_right


Output:




My Personal Notes arrow_drop_up

Check out this Author's contributed articles.

If you like GeeksforGeeks and would like to contribute, you can also write an article using contribute.geeksforgeeks.org or mail your article to contribute@geeksforgeeks.org. See your article appearing on the GeeksforGeeks main page and help other Geeks.

Please Improve this article if you find anything incorrect by clicking on the "Improve Article" button below.