Class SDLayerParams
- java.lang.Object
-
- org.deeplearning4j.nn.conf.layers.samediff.SDLayerParams
-
- All Implemented Interfaces:
Serializable
- Direct Known Subclasses:
SDVertexParams
public class SDLayerParams extends Object implements Serializable
- See Also:
- Serialized Form
-
-
Constructor Summary
Constructors Constructor Description SDLayerParams(Map<String,long[]> weightParams, Map<String,long[]> biasParams)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description void
addBiasParam(@NonNull String paramKey, @lombok.NonNull long... paramShape)
Add a bias parameter to the layer, with the specified shape.void
addWeightParam(@NonNull String paramKey, @lombok.NonNull long... paramShape)
Add a weight parameter to the layer, with the specified shape.void
clear()
Clear any previously set weight/bias parameters (including their shapes)boolean
equals(Object o)
List<String>
getBiasParameterKeys()
List<String>
getParameterKeys()
Map<String,long[]>
getParamShapes()
Get the parameter shapes for all parametersList<String>
getWeightParameterKeys()
int
hashCode()
boolean
isBiasParam(String param)
boolean
isWeightParam(String param)
-
-
-
Method Detail
-
addWeightParam
public void addWeightParam(@NonNull @NonNull String paramKey, @NonNull @lombok.NonNull long... paramShape)
Add a weight parameter to the layer, with the specified shape. For example, a standard fully connected layer could have weight parameters with shape [numInputs, layerSize]- Parameters:
paramKey
- The parameter key (name) for the weight parameterparamShape
- Shape of the weight parameter array
-
addBiasParam
public void addBiasParam(@NonNull @NonNull String paramKey, @NonNull @lombok.NonNull long... paramShape)
Add a bias parameter to the layer, with the specified shape. For example, a standard fully connected layer could have bias parameters with shape [1, layerSize]- Parameters:
paramKey
- The parameter key (name) for the bias parameterparamShape
- Shape of the bias parameter array
-
getParameterKeys
public List<String> getParameterKeys()
- Returns:
- Get a list of parameter names / keys (previously added via
addWeightParam(String, long...)
andaddBiasParam(String, long...)
-
getWeightParameterKeys
public List<String> getWeightParameterKeys()
- Returns:
- Get a list of parameter names / keys for weight parameters only, previously added via
addWeightParam(String, long...)
-
getBiasParameterKeys
public List<String> getBiasParameterKeys()
- Returns:
- Get a list of parameter names / keys for weight parameters only, previously added via
addWeightParam(String, long...)
-
getParamShapes
public Map<String,long[]> getParamShapes()
Get the parameter shapes for all parameters- Returns:
- Map of parameter shapes, by parameter
-
clear
public void clear()
Clear any previously set weight/bias parameters (including their shapes)
-
isWeightParam
public boolean isWeightParam(String param)
-
isBiasParam
public boolean isBiasParam(String param)
-
-