public abstract class SameDiffLambdaVertex extends SameDiffVertex
Modifier and Type | Class and Description |
---|---|
class |
SameDiffLambdaVertex.VertexInputs |
Modifier and Type | Field and Description |
---|---|
protected SameDiffLambdaVertex.VertexInputs |
inputs |
biasUpdater, dataType, gradientNormalization, gradientNormalizationThreshold, regularization, regularizationBias, updater
Constructor and Description |
---|
SameDiffLambdaVertex() |
Modifier and Type | Method and Description |
---|---|
GraphVertex |
clone() |
void |
defineParametersAndInputs(SDVertexParams params)
Define the parameters - and inputs - for the network.
|
SDVariable |
defineVertex(SameDiff sameDiff,
Map<String,SDVariable> layerInput,
Map<String,SDVariable> paramTable,
Map<String,SDVariable> maskVars)
Define the vertex
|
abstract SDVariable |
defineVertex(SameDiff sameDiff,
SameDiffLambdaVertex.VertexInputs inputs)
The defineVertex method is used to define the foward pass for the vertex
|
protected SameDiffLambdaVertex.VertexInputs |
getInputs(SameDiff sd) |
void |
initializeParameters(Map<String,INDArray> params)
Set the initial parameter values for this layer, if required
|
applyGlobalConfig, applyGlobalConfigToLayer, feedForwardMaskArrays, getGradientNormalization, getGradientNormalizationThreshold, getLayerName, getMemoryReport, getOutputType, getRegularizationByParam, getUpdaterByParam, getVertexParams, instantiate, isPretrainParam, maxVertexInputs, minVertexInputs, numParams, paramReshapeOrder, setDataType, validateInput
equals, hashCode
protected transient SameDiffLambdaVertex.VertexInputs inputs
public abstract SDVariable defineVertex(SameDiff sameDiff, SameDiffLambdaVertex.VertexInputs inputs)
sameDiff
- SameDiff instance to use to define the vertexinputs
- Layer input variablepublic SDVariable defineVertex(SameDiff sameDiff, Map<String,SDVariable> layerInput, Map<String,SDVariable> paramTable, Map<String,SDVariable> maskVars)
SameDiffVertex
defineVertex
in class SameDiffVertex
sameDiff
- SameDiff instancelayerInput
- Input to the layer - keys as defined by SameDiffVertex.defineParametersAndInputs(SDVertexParams)
paramTable
- Parameter table - keys as defined by SameDiffVertex.defineParametersAndInputs(SDVertexParams)
maskVars
- Masks of input, if available - keys as defined by SameDiffVertex.defineParametersAndInputs(SDVertexParams)
public void defineParametersAndInputs(SDVertexParams params)
SameDiffVertex
SDLayerParams.addWeightParam(String, long...)
and
SDLayerParams.addBiasParam(String, long...)
.
Note also you must define (and optionally name) the inputs to the vertex. This is required so that
DL4J knows how many inputs exists for the vertex.defineParametersAndInputs
in class SameDiffVertex
params
- Object used to set parameters for this layerpublic void initializeParameters(Map<String,INDArray> params)
SameDiffVertex
initializeParameters
in class SameDiffVertex
params
- Parameter arrays that may be initializedpublic GraphVertex clone()
clone
in class GraphVertex
protected SameDiffLambdaVertex.VertexInputs getInputs(SameDiff sd)
Copyright © 2022. All rights reserved.