我想知道深度神经网络中每个层的执行时间。有没有工具可以分析每个层的性能？我知道可以在模型中插入一些打印语句，但我不确定这样做是否准确。

回答：

要检查每个层的尺寸和参数，你可以使用torch summary。

要计算每个层的执行时间，你可以使用torchprof。

我不知道有哪个项目将这两个库合并在一起。也许这是你的机会（笑）