获取deeplearning4j java中每一层的输入和输出大小

Question

如何在dl4j中获取图层输入和输出大小？

例如，像这样的东西：

MultiLayerNetwork network = model.init()

for(Layer layer : network.getLayers()) {
 int[] outputShape = layer.shape() 
}

Answer 1

它比这更复杂一些，因为 DL4J 支持比简单的密集层或完全连接的层更复杂的层。

如果你想把那个信息打印出来，用起来可能更方便

String summary = model.summary();

如果您想对这些信息做些什么，您可以查看摘要方法本身的实现。

https://github.com/eclipse/deeplearning4j/blob/master/deeplearning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/multilayer/MultiLayerNetwork.java#L3636-L3757

特别是链接 3679 到 3699，因为它们都是关于获取层的输入和输出大小。