org.deeplearning4j.nn.conf.constraint

Class UnitNormConstraint

java.lang.Object

org.deeplearning4j.nn.conf.constraint.BaseConstraint

org.deeplearning4j.nn.conf.constraint.UnitNormConstraint

All Implemented Interfaces:

Serializable, Cloneable, LayerConstraint

public class UnitNormConstraint
extends BaseConstraint

Constrain the L2 norm of the incoming weights for each unit to be 1.0

Author:

Alex Black

See Also:

Serialized Form

Field Summary

Fields inherited from class org.deeplearning4j.nn.conf.constraint.BaseConstraint

DEFAULT_EPSILON, dimensions, epsilon, params

Constructor Summary

Constructors

Constructor and Description
UnitNormConstraint(int... dimensions) Apply to weights but not biases by default
UnitNormConstraint(Set<String> paramNames, int... dimensions)

Method Summary

Modifier and Type	Method and Description
void	apply(org.nd4j.linalg.api.ndarray.INDArray param)
UnitNormConstraint	clone()

Methods inherited from class org.deeplearning4j.nn.conf.constraint.BaseConstraint

applyConstraint, getBroadcastDims

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.deeplearning4j.nn.api.layers.LayerConstraint

getParams, setParams

Constructor Detail

UnitNormConstraint

public UnitNormConstraint(int... dimensions)

Apply to weights but not biases by default

Parameters:

dimensions - Dimensions to apply to. For DenseLayer, OutputLayer, RnnOutputLayer, LSTM, etc: this should be dimension 1. For CNNs, this should be dimensions [1,2,3] corresponding to last 3 of parameters which have order [depthOut, depthIn, kH, kW]

UnitNormConstraint

public UnitNormConstraint(Set<String> paramNames,
                          int... dimensions)

Parameters:

dimensions - Dimensions to apply to. For DenseLayer, OutputLayer, RnnOutputLayer, LSTM, etc: this should be dimension 1. For CNNs, this should be dimensions [1,2,3] corresponding to last 3 of parameters which have order [depthOut, depthIn, kH, kW]

Method Detail

apply

public void apply(org.nd4j.linalg.api.ndarray.INDArray param)

Specified by:

apply in class BaseConstraint

clone

public UnitNormConstraint clone()

Specified by:

clone in interface LayerConstraint

Specified by:

clone in class BaseConstraint