繁体   English   中英

Tensorflow:为什么tf.nn.conv2d运行得比tf.layers.conv2d快?

[英]Tensorflow: why tf.nn.conv2d runs faster than tf.layers.conv2d?

我正在编写一个简单的AlexNet实现。 我尝试使用tf.nn.conv2d和tf.layers.conv2d,结果表明,使用tf.nn.conv2d时,损失的下降速度更快,即使结构完全相同。 有谁知道任何解释吗?

如果你试图遵循的函数调用链中,你会发现, tf.layers.conv2D()将调用tf.nn.conv2D()所以无论你使用什么, tf.nn.conv2d()被调用,如果您自己命名,它将更快。 您可以使用traceback.print_stack()方法亲自进行验证。

注意这并不意味着它们是相同的,请根据您的需要选择功能,因为tf.layers.conv2D()承担其他各种任务。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM