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()。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。