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.

Author:

Justin Long (crockpotveggies)

Method Summary

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

Method Detail

getModuleName

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,
                                                         String layerName,
                                                         int inputSize,
                                                         int[][] config,
                                                         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.