public abstract class SameDiffOutputLayer extends AbstractSameDiffLayer
AbstractSameDiffLayer.Builder<T extends AbstractSameDiffLayer.Builder<T>>biasUpdater, gradientNormalization, gradientNormalizationThreshold, regularization, regularizationBias, updaterconstraints, iDropout, layerName| Modifier | Constructor and Description |
|---|---|
protected |
SameDiffOutputLayer() |
| Modifier and Type | Method and Description |
|---|---|
abstract String |
activationsVertexName()
Output layers should terminate in a single scalar value (i.e., a score) - however, sometimes the output activations
(such as softmax probabilities) need to be returned.
|
abstract SDVariable |
defineLayer(SameDiff sameDiff,
SDVariable layerInput,
SDVariable labels,
Map<String,SDVariable> paramTable)
Define the output layer
|
Layer |
instantiate(NeuralNetConfiguration conf,
Collection<TrainingListener> trainingListeners,
int layerIndex,
INDArray layerParamsView,
boolean initializeParams,
DataType networkDataType) |
boolean |
labelsRequired()
Whether labels are required for calculating the score.
|
applyGlobalConfig, applyGlobalConfigToLayer, defineParameters, getLayerParams, getMemoryReport, getPreProcessorForInputType, getRegularizationByParam, getUpdaterByParam, initializeParameters, initializer, initWeights, isPretrainParam, onesMaskForInput, paramReshapeOrder, setNInclone, getOutputType, initializeConstraints, resetLayerDefaultConfig, setDataTypeequals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, waitgetGradientNormalization, getGradientNormalizationThreshold, getLayerNamepublic abstract SDVariable defineLayer(SameDiff sameDiff, SDVariable layerInput, SDVariable labels, Map<String,SDVariable> paramTable)
sameDiff - SameDiff instancelayerInput - Input to the layerlabels - Labels variable (or null if labelsRequired() returns falseparamTable - Parameter table - keys as defined by AbstractSameDiffLayer.defineParameters(SDLayerParams)public abstract String activationsVertexName()
public boolean labelsRequired()
public Layer instantiate(NeuralNetConfiguration conf, Collection<TrainingListener> trainingListeners, int layerIndex, INDArray layerParamsView, boolean initializeParams, DataType networkDataType)
instantiate in class AbstractSameDiffLayerCopyright © 2022. All rights reserved.