神經網絡前向傳播和反向傳播

兩層神經網絡 (N,D_I,H,D_O)(64,10000,1000,10) 公式(簡化沒放b) H=w1x y=w2H #設置learning_ratw=1e-6 #數據初始化 x,y,w,b #前向傳播 把公式的內容寫一遍 #計算loss值(y_pred-y) #反向傳播 從loss裏得到grad_y_pred--->grad_w2--->grad_h_relu--->grad_h--->gr
相關文章
相關標籤/搜索