Tensorflow:为什么tf.nn.conv2d型比…跑得快tf.layers.conv2d?

2024-05-16 12:40:27 发布

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

我正在编写AlexNet的一个简单实现。我试过用tf.nn.conv2d型以及tf.layers.conv2d文件结果表明,在使用时,损耗下降较快tf.nn.conv2d型,甚至结构也完全相同。有人知道这有什么解释吗?你知道吗


Tags: 文件layerstfnn结构损耗conv2d结果表明
1条回答
网友
1楼 · 发布于 2024-05-16 12:40:27

如果你试着沿着函数调用链,你会发现tf.layers.conv2D()调用tf.nn.conv2D(),所以不管你用什么,都会被调用tf.nn.conv2d(),如果你自己调用它会更快。您可以使用traceback.print_stack()方法自己验证。你知道吗

注意这并不意味着它们是一个相同的函数,请根据需要选择函数,因为tf.layers.conv2D()还承担着其他各种任务。你知道吗

相关问题 更多 >