在Tensorflow中實現新操作時使用簡單的數學函數
[英]Using simple math functions when implementing new op in Tensorflow
問題描述
我正在嘗試通過編寫.cc文件（CPU和GPU版本）在C ++中實現新的操作。 該操作使用簡單的數學函數，例如cos或sin。
在CPU / GPU的浮點數上調用cos或sin（或類似）的正確方法是什么。 我們是否必須包括一個外部庫？ 是在Eigen嗎？ 我們必須為CPU和GPU使用不同的庫嗎？ 怎么申報呢？
除了zero_out.cc之外，還有其他一些文檔/示例嗎？
1 個解決方案
解決方案1
0 2017-06-02 16:17:49
我使用了std::cos和std::sin ，它在CPU上工作正常！ 不知道這是怎么回事，我會一直保留這個答案，直到有人反對為止。
編輯：嗯，看來我講得太快了：GPU版本需要CUDA函數，而不是C ++函數（所謂的主機函數）。
使用 TPU 時抑制 Tensorflow “Executing op”消息
[英]Suppress Tensorflow "Executing op" messages when using TPU
在 Tensorflow 中，我是否需要為“sinc”或“gaussian”激活函數添加新操作？
[英]In Tensorflow, do I need to add new op for "sinc" or "gaussian" activation functions?
如何根據現有Tensorflow操作組成來創建新的Tensorflow操作
[英]How to create new Tensorflow op from composition of existing Tensorflow ops
AttributeError: 當在 Tensorflow federated 中使用多個特征輸入時啟用急切執行時，Tensor.op 毫無意義
[英]AttributeError: Tensor.op is meaningless when eager execution is enabled when using multiple feature inputs in Tensorflow federated
聲明:本站的技術帖子網頁，遵循CC BY-SA 4.0協議，如果您需要轉載，請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.
相關問題
在tensorflow中添加新的op-形狀函數
在張量流中導入圖形時使用新的op
使用 TPU 時抑制 Tensorflow “Executing op”消息
在TensorFlow中定義自定義漸變時使用運算輸入
在 Tensorflow 中，我是否需要為“sinc”或“gaussian”激活函數添加新操作？
Tensorflow vs Numpy數學函數
編譯new_op教程時出錯（Tensorflow）
如何根據現有Tensorflow操作組成來創建新的Tensorflow操作
AttributeError: 當在 Tensorflow federated 中使用多個特征輸入時啟用急切執行時，Tensor.op 毫無意義
Tensorflow添加了一個新的op，無法從python導入
相關標簽