實現屬於自己的TensorFlow(一) - 計算圖與前向傳播

前段時間因爲課題需要使用了一段時間TensorFlow,感覺這種框架很有意思,除了可以搭建複雜的神經網絡,也可以優化其他自己需要的計算模型,所以一直想自己學習一下寫一個類似的圖計算框架。前幾天組會開完決定着手實現一個模仿TensorFlow接口的簡陋版本圖計算框架以學習計算圖程序的編寫以及前向傳播和反向傳播的實現。目前實現了前向傳播和反向傳播以及梯度下降優化器,並寫了個優化線性模型的例子。廈門叉車
相關文章
相關標籤/搜索