获取deeplearning4java中各层的输入和输出大小

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

例如，像这样：

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

回答：

这比看起来要复杂一些，因为DL4J支持的层比简单的密集层或全连接层要复杂得多。

如果你只是想打印这些信息，使用下面的方法可能会更简单：

String summary = model.summary();

如果你想利用这些信息做一些事情，你可以查看summary方法本身的实现。

特别是第3679到3699行，这些行都与获取层的输入和输出大小有关。

学技术