Python tensorflow.math.square()用法及代码示例

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

square()用于计算x的元素明智平方，即x * x。

用法：tensorflow.math.square(x, name)

参数：

x:是张量允许的dtypes是bfloat16，half，float32，float64，complex64，complex128。
name(optional):它定义了操作的名称。

返回值：它返回一个张量。

范例1：

Python3

# 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：可视化

Python3

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

输出：

相关用法

注：本文由纯净天空筛选整理自aman neekhara大神的英文原创作品 Python – tensorflow.math.square()。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。