获取deeplearning4java中每个图层的输入和输出大小

import org.deeplearning4j.nn.api.Layer;
import org.deeplearning4j.nn.conf.MultiLayerConfiguration;
import org.deeplearning4j.nn.conf.NeuralNetConfiguration;
import org.deeplearning4j.nn.multilayer.MultiLayerNetwork;
import org.deeplearning4j.nn.weights.WeightInit;

NeuralNetConfiguration.Builder builder = new NeuralNetConfiguration.Builder();
builder.weightInit(WeightInit.XAVIER);
// 添加其他配置参数，如激活函数、优化器等

MultiLayerConfiguration conf = builder.list()
    // 添加图层配置，如DenseLayer、ConvolutionLayer等
    .layer(...)
    .layer(...)
    .layer(...)
    .build();

MultiLayerNetwork model = new MultiLayerNetwork(conf);
model.init();

for (Layer layer : model.getLayers()) {
    int inputSize = layer.input().shape()[1]; // 输入大小
    int outputSize = layer.output().shape()[1]; // 输出大小
    System.out.println("图层输入大小：" + inputSize);
    System.out.println("图层输出大小：" + outputSize);
}