我正在寻找TensorFlow中math_ops.sin
和math_ops.cos
函数的实现。他们内部使用numpy吗?
答案 0 :(得分:0)
您是在谈论tf.cos
(https://www.tensorflow.org/api_docs/python/tf/cos)和tf.sin
吗?
答案 1 :(得分:0)
它们在C ++接口中定义,在内部使用Eigen或CUDA库(取决于CUDA是否启用)。
Here您可以看到余弦函数的定义位置。
在math_ops.cc中,您可以看到" Cos"功能已注册。注册后,可以从python导入,从math_ops
模块导入它。