public abstract class SameDiffLayer extends AbstractSameDiffLayer
SameDiffOutputLayer
instead.SameDiffVertex
Modifier and Type | Class and Description |
---|---|
static class |
SameDiffLayer.Builder<T extends SameDiffLayer.Builder<T>> |
Modifier and Type | Field and Description |
---|---|
protected WeightInit |
weightInit |
biasUpdater, gradientNormalization, gradientNormalizationThreshold, regularization, regularizationBias, updater
constraints, iDropout, layerName
Modifier | Constructor and Description |
---|---|
protected |
SameDiffLayer() |
protected |
SameDiffLayer(SameDiffLayer.Builder builder) |
Modifier and Type | Method and Description |
---|---|
abstract SDVariable |
defineLayer(SameDiff sameDiff,
SDVariable layerInput,
Map<String,SDVariable> paramTable,
SDVariable mask)
Define the layer
|
Pair<INDArray,MaskState> |
feedForwardMaskArray(INDArray maskArray,
MaskState currentMaskState,
int minibatchSize) |
Layer |
instantiate(NeuralNetConfiguration conf,
Collection<TrainingListener> trainingListeners,
int layerIndex,
INDArray layerParamsView,
boolean initializeParams,
org.nd4j.linalg.api.buffer.DataType networkDataType) |
void |
validateInput(INDArray input)
Validate input arrays to confirm that they fulfill the assumptions of the layer.
|
applyGlobalConfig, applyGlobalConfigToLayer, defineParameters, getLayerParams, getMemoryReport, getPreProcessorForInputType, getRegularizationByParam, getUpdaterByParam, initializeParameters, initializer, initWeights, isPretrainParam, onesMaskForInput, paramReshapeOrder, setNIn
clone, getOutputType, initializeConstraints, resetLayerDefaultConfig, setDataType
equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
getGradientNormalization, getGradientNormalizationThreshold, getLayerName
protected WeightInit weightInit
protected SameDiffLayer(SameDiffLayer.Builder builder)
protected SameDiffLayer()
public abstract SDVariable defineLayer(SameDiff sameDiff, SDVariable layerInput, Map<String,SDVariable> paramTable, SDVariable mask)
sameDiff
- SameDiff instancelayerInput
- Input to the layerparamTable
- Parameter table - keys as defined by AbstractSameDiffLayer.defineParameters(SDLayerParams)
mask
- Optional, maybe null. Mask to apply if supportedpublic Pair<INDArray,MaskState> feedForwardMaskArray(INDArray maskArray, MaskState currentMaskState, int minibatchSize)
public void validateInput(INDArray input)
input
- input to the layerpublic Layer instantiate(NeuralNetConfiguration conf, Collection<TrainingListener> trainingListeners, int layerIndex, INDArray layerParamsView, boolean initializeParams, org.nd4j.linalg.api.buffer.DataType networkDataType)
instantiate
in class AbstractSameDiffLayer
Copyright © 2019. All rights reserved.