org.deeplearning4j.nn.modelimport.keras.layers.pooling

Class KerasPooling3D

java.lang.Object

org.deeplearning4j.nn.modelimport.keras.KerasLayer

org.deeplearning4j.nn.modelimport.keras.layers.pooling.KerasPooling3D

public class KerasPooling3D
extends KerasLayer

Imports a Keras 3D Pooling layer as a DL4J Subsampling3D layer.

Author:

Max Pumperla

Nested Class Summary

Nested classes/interfaces inherited from class org.deeplearning4j.nn.modelimport.keras.KerasLayer

KerasLayer.DimOrder

Field Summary

Fields inherited from class org.deeplearning4j.nn.modelimport.keras.KerasLayer

className, conf, dimOrder, dropout, inboundLayerNames, inputShape, kerasMajorVersion, layer, layerName, vertex, weightL1Regularization, weightL2Regularization, weights

Constructor Summary

Constructors

Constructor and Description
KerasPooling3D(Map<String,Object> layerConfig) Constructor from parsed Keras layer configuration dictionary.
KerasPooling3D(Map<String,Object> layerConfig, boolean enforceTrainingConfig) Constructor from parsed Keras layer configuration dictionary.

Method Summary

Modifier and Type	Method and Description
InputType	getOutputType(InputType... inputType) Get layer output type.
Subsampling3DLayer	getSubsampling3DLayer() Get DL4J Subsampling3DLayer.

Methods inherited from class org.deeplearning4j.nn.modelimport.keras.KerasLayer

clearCustomLayers, clearLambdaLayers, copyWeightsToLayer, getClassName, getDimOrder, getInboundLayerNames, getInputPreprocessor, getInputShape, getKerasMajorVersion, getLayer, getLayerName, getNInFromConfig, getNumParams, getVertex, getWeights, isInputPreProcessor, isLayer, isValidInboundLayer, isVertex, registerCustomLayer, registerLambdaLayer, setInboundLayerNames, setWeights, usesRegularization

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

KerasPooling3D

public KerasPooling3D(Map<String,Object> layerConfig)
               throws InvalidKerasConfigurationException,
                      UnsupportedKerasConfigurationException

Constructor from parsed Keras layer configuration dictionary.

Parameters:

layerConfig - dictionary containing Keras layer configuration.

Throws:

InvalidKerasConfigurationException - Invalid Keras config

UnsupportedKerasConfigurationException - Unsupported Keras config

KerasPooling3D

public KerasPooling3D(Map<String,Object> layerConfig,
                      boolean enforceTrainingConfig)
               throws InvalidKerasConfigurationException,
                      UnsupportedKerasConfigurationException

Constructor from parsed Keras layer configuration dictionary.

Parameters:

layerConfig - dictionary containing Keras layer configuration

enforceTrainingConfig - whether to enforce training-related configuration options

Throws:

InvalidKerasConfigurationException - Invalid Keras config

UnsupportedKerasConfigurationException - Unsupported Keras config

Method Detail

getSubsampling3DLayer

public Subsampling3DLayer getSubsampling3DLayer()

Get DL4J Subsampling3DLayer.

Returns:

Subsampling3DLayer

getOutputType

public InputType getOutputType(InputType... inputType)
                        throws InvalidKerasConfigurationException

Get layer output type.

Overrides:

getOutputType in class KerasLayer

Parameters:

inputType - Array of InputTypes

Returns:

output type as InputType

Throws:

InvalidKerasConfigurationException - Invalid Keras config