TensorFlow是Google設計的開源Python庫,用於開發機器學習模型和深度學習神經網絡。
gradients()用於獲取ys w.r.t之和的符號導數。 x在xs中。啟用急切執行後,此函數將無效。
Syantx:tensorflow.gradients(ys,xs,grad_ys,名稱,gate_gradients,aggregation_method,stop_gradients,unconnected_gradients)
參數:
- ys:它是需要區分的張量或張量列表。
- xs:它是用於區分的張量或張量列表。
- grad_ys(可選):它是張量或張量列表,用於計算y的梯度。
- name(optional):一起使用組梯度運算。默認值為漸變。
- gate_gradients(可選):它用來避免比賽條件。如果為true,它將在為操作返回的漸變周圍添加一個元組。
- aggregation_method(可選):它的值是在AggregationMethod類中定義的常量。
- stop_gradients(可選):這是張量或張量列表,無法區分。
- unconnected_gradients(可選):它指定當給定的輸入張量未連接時返回的梯度值。接受的值是在UnconnectedGradients類中定義的常量。
返回值:長度為len(xs)的張量的列表,其中每個張量是ys中y和xs中x的總和(dy /dx)。
範例1:
Python3
# Importing the library
import tensorflow as tf
# Defining function
@tf.function
def gfg():
a = tf.ones([1, 2])
b = 5*a
# Calculaing gradient
g1 = tf.gradients([b+a], [a])
# Printing result
print("res:",g1)
# Calling the function
gfg()
輸出:
res: [<tf.Tensor 'gradients/AddN:0' shape=(1, 2) dtype=float32>]
範例2:
Python3
# Importing the library
import tensorflow as tf
# Defining function
@tf.function
def gfg():
a = tf.ones([1, 2])
b = 5*a
# Calculaing gradient
g1 = tf.gradients([b], [a])
# Printing result
print("res:",g1)
# Calling the function
gfg()
輸出:
res: [<tf.Tensor 'gradients/mul_grad/Mul_1:0' shape=(1, 2) dtype=float32>]
相關用法
注:本文由純淨天空篩選整理自aman neekhara大神的英文原創作品 Python – tensorflow.gradients()。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。