org.deeplearning4j.nn.conf.module

Interface GraphBuilderModule

public interface GraphBuilderModule

GraphBuilderModule for nn layers. Allows for creation of plugins and modules to generate configurations and layers.

Method Summary

Modifier and Type	Method and Description
java.lang.String	getModuleName() A module should return its name.
ComputationGraphConfiguration.GraphBuilder	updateBuilder(ComputationGraphConfiguration.GraphBuilder graph, java.lang.String layerName, int inputSize, int[][] config, java.lang.String inputLayer) Add a layer to the collection of layers being generated by this module.

Method Detail

getModuleName

java.lang.String getModuleName()

A module should return its name. Useful for automatically generating layer names.

Returns:

The module name as a string.

updateBuilder

ComputationGraphConfiguration.GraphBuilder updateBuilder(ComputationGraphConfiguration.GraphBuilder graph,
                                                         java.lang.String layerName,
                                                         int inputSize,
                                                         int[][] config,
                                                         java.lang.String inputLayer)

Add a layer to the collection of layers being generated by this module.

Parameters:

graph -

layerName -

inputSize -

config -

inputLayer -

Returns:

An instance of GraphBuilder with appended layers.