TensorFlow中math_ops三角函数的实现在哪里?

时间:2017-10-30 06:19:59

标签: tensorflow

我正在寻找TensorFlow中math_ops.sinmath_ops.cos函数的实现。他们内部使用numpy吗?

2 个答案:

答案 0 :(得分:0)

您是在谈论tf.coshttps://www.tensorflow.org/api_docs/python/tf/cos)和tf.sin吗?

答案 1 :(得分:0)

它们在C ++接口中定义,在内部使用Eigen或CUDA库(取决于CUDA是否启用)。

Here您可以看到余弦函数的定义位置。

math_ops.cc中,您可以看到" Cos"功能已注册。注册后,可以从python导入,从math_ops模块导入它。