[英]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.