.addVertex("merge", new MergeVertex(), "cnn3", "cnn4", "cnn5") //Perform depth concatenation .addLayer("globalPool", new GlobalPoolingLayer.Builder() .poolingType(globalPoolingType) .build(), "merge") .addLayer("out", new OutputLayer.Builder()
@Override public GlobalPoolingLayer getValue(double[] parameterValues) { GlobalPoolingLayer.Builder builder = new GlobalPoolingLayer.Builder(); super.setLayerOptionsBuilder(builder, parameterValues); if (poolingDimensions != null) builder.poolingDimensions(poolingDimensions.getValue(parameterValues)); if (collapseDimensions != null) builder.collapseDimensions(collapseDimensions.getValue(parameterValues)); if (poolingType != null) builder.poolingType(poolingType.getValue(parameterValues)); if (pNorm != null) builder.pnorm(pNorm.getValue(parameterValues)); return builder.build(); }