您能指出我tf.nn.sigmoid在TensorFlow（2.0）中该函数的实际c ++实现吗？仅出于教育目的。
谢谢。
Tensorflow源代码可能很难导航。这是我用来查找实现文件的一些想法。
您可以tf.nn.sigmoid在这里看到在python中定义的
tensorflow /蟒蛇/ OPS / math_ops.py。
如果查看此函数定义，可以看到它返回了中定义的S型函数gen_math_ops。我花了一段时间来弄清楚，但如果你去寻找gen_math_ops在tensorflow/python/ops你不会找到它。gen_tensorflow的名称前面带有任何名称，该名称由C ++代码注册。
我们真正想要的是内核实现。这些可以在中找到tensorflow/core/kernels。快速ctrl + f用于将"sigmoid"我们指向cwise_op_sigmoid.cc。它不包含实现，但将我们指向头文件cwise_ops_common.h。这也不包含实现，但将我们指向cwise_ops.h。"sigmoid"在此文件中使用ctrl + f ，我们在第877行找到了“实现” 。您可以看到这是包装Eigen操作的函子Eigen::internal::scalar_logistic_op。这是该操作的文档。如果您想知道如何实现该操作，请下载源代码。你可以在这里找到它。
希望这可以帮助。
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句