weka.dl4j.layers.lambda

Class CustomBroadcast

java.lang.Object

org.deeplearning4j.nn.conf.layers.Layer

org.deeplearning4j.nn.conf.layers.samediff.AbstractSameDiffLayer

org.deeplearning4j.nn.conf.layers.samediff.SameDiffLayer

org.deeplearning4j.nn.conf.layers.samediff.SameDiffLambdaLayer

weka.dl4j.layers.lambda.CustomBroadcast

All Implemented Interfaces:

java.io.Serializable, java.lang.Cloneable, org.deeplearning4j.nn.api.TrainingConfig

public class CustomBroadcast
extends org.deeplearning4j.nn.conf.layers.samediff.SameDiffLambdaLayer

Required for loading the EfficientNet family of models. Simply broadcasts the activations up to the correct size for the ElementWiseVertex to multiply the activations.

See Also:

Serialized Form

Nested Class Summary

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

org.deeplearning4j.nn.conf.layers.samediff.SameDiffLayer.Builder<T extends org.deeplearning4j.nn.conf.layers.samediff.SameDiffLayer.Builder<T>>

Constructor Summary

Constructors

Constructor and Description
CustomBroadcast() Instantiate the layer.
CustomBroadcast(long width) Instantiate the layer with the supplied width.

Method Summary

Modifier and Type	Method and Description
org.nd4j.autodiff.samediff.SDVariable	defineLayer(org.nd4j.autodiff.samediff.SameDiff sd, org.nd4j.autodiff.samediff.SDVariable x)
org.deeplearning4j.nn.conf.inputs.InputType	getOutputType(int layerIndex, org.deeplearning4j.nn.conf.inputs.InputType inputType)

Methods inherited from class org.deeplearning4j.nn.conf.layers.samediff.SameDiffLambdaLayer

defineLayer, defineParameters, initializeParameters

Methods inherited from class org.deeplearning4j.nn.conf.layers.samediff.SameDiffLayer

equals, feedForwardMaskArray, getParamWeightInit, getWeightInit, hashCode, instantiate, setParamWeightInit, setWeightInit, toString, validateInput

Methods inherited from class org.deeplearning4j.nn.conf.layers.samediff.AbstractSameDiffLayer

applyGlobalConfig, applyGlobalConfigToLayer, getBiasUpdater, getGradientNormalization, getGradientNormalizationThreshold, getLayerParams, getMemoryReport, getPreProcessorForInputType, getRegularization, getRegularizationBias, getRegularizationByParam, getUpdater, getUpdaterByParam, initializer, isPretrainParam, onesMaskForInput, paramReshapeOrder, setBiasUpdater, setGradientNormalization, setGradientNormalizationThreshold, setLayerParams, setNIn, setRegularization, setRegularizationBias, setUpdater

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

clone, getConstraints, getIDropout, getLayerName, resetLayerDefaultConfig, setConstraints, setDataType, setIDropout, setLayerName

Methods inherited from class java.lang.Object

getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

CustomBroadcast

public CustomBroadcast()

Instantiate the layer.

CustomBroadcast

public CustomBroadcast(long width)

Instantiate the layer with the supplied width.

Parameters:

width - Desired Width

Method Detail

defineLayer

public org.nd4j.autodiff.samediff.SDVariable defineLayer(org.nd4j.autodiff.samediff.SameDiff sd,
                                                         org.nd4j.autodiff.samediff.SDVariable x)

Specified by:

defineLayer in class org.deeplearning4j.nn.conf.layers.samediff.SameDiffLambdaLayer

getOutputType

public org.deeplearning4j.nn.conf.inputs.InputType getOutputType(int layerIndex,
                                                                 org.deeplearning4j.nn.conf.inputs.InputType inputType)

Overrides:

getOutputType in class org.deeplearning4j.nn.conf.layers.samediff.SameDiffLambdaLayer