Tensorflow:tf.nn.conv2d实际执行在哪里?

2024-05-23 19:11:14 发布

您现在位置:Python中文网/ 问答频道 /正文

我对Tensorflow实现tf.nn.conv2d(...)很好奇。要调用它,只需运行tf.nn.conv2d(...)。不过,我要去兔子洞看看它是在哪里被处决的。代码如下(箭头表示它最终调用的函数):

tf.nn.conv2d(...) -> tf.nn_ops.conv2d(...) -> tf.gen_nn_ops.conv2d(...) -> _op_def_lib.apply_op("Conv2D", ...) -> ?

我熟悉Tensorflow对LSTMs的实现,以及在人们认为合适的情况下轻松操作它们的能力。执行conv2d()计算的函数是用Python编写的吗?如果是,它在哪里?我能看看这些步伐是在哪里以及如何执行的吗?


Tags: 函数代码libtftensorflowdefnn箭头