我在寻找实施math_ops.sin和math_ops.cos功能TensorFlow。他们在内部使用 numpy 吗？
它们在 C++ 接口中定义，并且在内部使用 Eigen 或 CUDA 库（这取决于是否启用了 CUDA）。
在这里您可以看到余弦函数的定义位置。
相反，在math_ops.cc 中，您可以看到“Cos”函数的注册位置。注册后，它可以从 python 导入，从math_ops模块导入它。
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句