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, validateInputequals, hashCodeprotected 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)
SameDiffVertexdefineVertex in class SameDiffVertexsameDiff - 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)
SameDiffVertexSDLayerParams.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 SameDiffVertexparams - Object used to set parameters for this layerpublic void initializeParameters(Map<String,INDArray> params)
SameDiffVertexinitializeParameters in class SameDiffVertexparams - Parameter arrays that may be initializedpublic GraphVertex clone()
clone in class GraphVertexprotected SameDiffLambdaVertex.VertexInputs getInputs(SameDiff sd)
Copyright © 2021. All rights reserved.