過去
左手程式右手詩?
2017年6月18日 星期日
tensorflow 自訂 operator
考慮開發速度的話,先使用
tf.py_func
[0]
如果遇到限制,或是考慮執行效能的話,再去看
Adding a New Op
[1]
雖說 [1] 比 [0] 麻煩,但其實也相當簡單,不用害怕,進去看著照做就可以了,半個小時的事情
[0]
https://www.tensorflow.org/api_docs/python/tf/py_func
[1]
https://www.tensorflow.org/extend/adding_an_op
沒有留言:
張貼留言
較新的文章
較舊的文章
首頁
訂閱:
張貼留言 (Atom)
沒有留言:
張貼留言