org.deeplearning4j.nn.conf.layers.samediff

Class SDVertexParams

java.lang.Object

org.deeplearning4j.nn.conf.layers.samediff.SDLayerParams

org.deeplearning4j.nn.conf.layers.samediff.SDVertexParams

All Implemented Interfaces:

Serializable

public class SDVertexParams
extends SDLayerParams

SDVertexParams is used to define the inputs - and the parameters - for a SameDiff vertex

Author:

Alex Black

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected List<String>	inputs

Constructor Summary

Constructors

Constructor and Description
SDVertexParams()

Method Summary

Modifier and Type	Method and Description
void	defineInputs(int numInputs) Define the inputs to the DL4J SameDiff vertex with generated names.
void	defineInputs(String... inputNames) Define the inputs to the DL4J SameDiff Vertex with specific names

Methods inherited from class org.deeplearning4j.nn.conf.layers.samediff.SDLayerParams

addBiasParam, addWeightParam, clear, equals, getBiasParameterKeys, getParameterKeys, getParamShapes, getWeightParameterKeys, hashCode, isBiasParam, isWeightParam

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, toString, wait, wait, wait

Field Detail

inputs

protected List<String> inputs

Constructor Detail

SDVertexParams

public SDVertexParams()

Method Detail

defineInputs

public void defineInputs(String... inputNames)

Define the inputs to the DL4J SameDiff Vertex with specific names

Parameters:

inputNames - Names of the inputs. Number here also defines the number of vertex inputs

See Also:

defineInputs(int)

defineInputs

public void defineInputs(int numInputs)

Define the inputs to the DL4J SameDiff vertex with generated names. Names will have format "input_0", "input_1", etc

Parameters:

numInputs - Number of inputs to the vertex.