溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》
您好，登錄后才能下訂單哦！
Tensorflow2.0中如何自動求梯度，很多新手對此不是很清楚，為了幫助大家解決這個難題，下面小編將為大家詳細講解，有這方面需求的人可以來學(xué)習(xí)下，希望你能有所收獲。
對于給定函數(shù)：y(w)=aw^2+bw+c
數(shù)學(xué)求導(dǎo)得：dy/dw=2aw+b
那么，（a,b,c,w）=(1,2,3,4)處的導(dǎo)數(shù)，dy/dw=2 * 1 *4 + 2=10
而在Tensorflow2.0中，梯度可以自動求取。具體代碼如下：
import tensorflow as tf a=tf.constant(1.)b=tf.constant(2.)c=tf.constant(3.)w=tf.constant(4.)with tf.GradientTape() as tape: # 構(gòu)建梯度環(huán)境tape.watch([w]) # 對w求梯度，并跟蹤列表y=a*w**2+b*w+c # 構(gòu)建目標函數(shù)[dy_dw] = tape.gradient(y, [w]) # 函數(shù)y對w求導(dǎo)print(dy_dw) # 打印出導(dǎo)數(shù)
運行結(jié)果：
看完上述內(nèi)容是否對您有幫助呢？如果還想對相關(guān)知識有進一步的了解或閱讀更多相關(guān)文章，請關(guān)注億速云行業(yè)資訊頻道，感謝您對億速云的支持。
免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點不代表本網(wǎng)站立場，如果涉及侵權(quán)請聯(lián)系站長郵箱：is@yisu.com進行舉報，并提供相關(guān)證據(jù)，一經(jīng)查實，將立刻刪除涉嫌侵權(quán)內(nèi)容。
億速云公眾號
手機網(wǎng)站二維碼
