我目前正在Keras中实现AlexNet。最近，我在一个仓库中看到了一个相当简洁的AlexNet架构实现。但是我觉得这个代码行中可能存在一些错误。如果我能找到一种方法来打印变量的维度，那么我就能理解并解决这个错误。我尝试过使用print函数和写入文件，但这些方法似乎都不起作用。有人知道如何从Keras库的这种自定义层中打印输出吗？

回答：

尝试使用`model.summary()`函数。它会打印出你的层维度的完整信息。