org.deeplearning4j.nn.modelimport.keras.layers

Class TFOpLayer

java.lang.Object

org.deeplearning4j.nn.conf.layers.Layer

org.deeplearning4j.nn.modelimport.keras.layers.TFOpLayer

All Implemented Interfaces:

Serializable, Cloneable, TrainingConfig

public class TFOpLayer
extends Layer

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class org.deeplearning4j.nn.conf.layers.Layer

Layer.Builder<T extends Layer.Builder<T>>

Field Summary

Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer

constraints, iDropout, layerName

Constructor Summary

Constructors

Constructor and Description
TFOpLayer(Map nodeDef, Map constants)

Method Summary

Modifier and Type	Method and Description
GradientNormalization	getGradientNormalization()
double	getGradientNormalizationThreshold()
LayerMemoryReport	getMemoryReport(InputType inputType)
InputType	getOutputType(int idx, InputType inputType)
InputPreProcessor	getPreProcessorForInputType(InputType inputType)
List<Regularization>	getRegularizationByParam(String paramName)
ParamInitializer	initializer()
Layer	instantiate(NeuralNetConfiguration conf, Collection<TrainingListener> trainingListeners, int layerIndex, INDArray layerParamsView, boolean initializeParams, DataType networkDataType)
boolean	isPretrainParam(String param)
void	setNIn(InputType inputType, boolean override)

Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer

canEqual, clone, equals, getConstraints, getIDropout, getLayerName, getUpdaterByParam, hashCode, initializeConstraints, resetLayerDefaultConfig, setConstraints, setDataType, setIDropout, setLayerName, toString

Methods inherited from class java.lang.Object

finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

TFOpLayer

public TFOpLayer(Map nodeDef,
                 Map constants)

Method Detail

initializer

public ParamInitializer initializer()

Specified by:

initializer in class Layer

getPreProcessorForInputType

public InputPreProcessor getPreProcessorForInputType(InputType inputType)

Specified by:

getPreProcessorForInputType in class Layer

isPretrainParam

public boolean isPretrainParam(String param)

Specified by:

isPretrainParam in interface TrainingConfig

Specified by:

isPretrainParam in class Layer

getOutputType

public InputType getOutputType(int idx,
                               InputType inputType)

Specified by:

getOutputType in class Layer

setNIn

public void setNIn(InputType inputType,
                   boolean override)

Specified by:

setNIn in class Layer

getGradientNormalization

public GradientNormalization getGradientNormalization()

instantiate

public Layer instantiate(NeuralNetConfiguration conf,
                         Collection<TrainingListener> trainingListeners,
                         int layerIndex,
                         INDArray layerParamsView,
                         boolean initializeParams,
                         DataType networkDataType)

Specified by:

instantiate in class Layer

getGradientNormalizationThreshold

public double getGradientNormalizationThreshold()

getRegularizationByParam

public List<Regularization> getRegularizationByParam(String paramName)

Specified by:

getRegularizationByParam in interface TrainingConfig

Specified by:

getRegularizationByParam in class Layer

getMemoryReport

public LayerMemoryReport getMemoryReport(InputType inputType)

Specified by:

getMemoryReport in class Layer