Python – tensorflow.math.square()

TensorFlow是谷歌设计的开源Python库，用于开发机器学习模型和深度学习神经网络。

square()是用来计算x的元素的平方，即x*x。

语法:
tensorflow.math.square(x, name)

参数:

• x:它是一个张量。允许的d类型是bfloat16, half, float32, float64, complex64, complex128。
• name（可选）:它定义了该操作的名称。

返回：
它返回一个张量。

示例 1:

# 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)

输出:

a: tf.Tensor([-5. -7. 2. 5. 7.], shape=(5, ), dtype=float64)
Result: tf.Tensor([25. 49. 4. 25. 49.], shape=(5, ), dtype=float64)

例子2：
视觉化

# 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()

输出:

")