public enum WeightInit extends Enum<WeightInit>
Modifier and Type | Method and Description |
---|---|
IWeightInit |
getWeightInitFunction()
Create an instance of the weight initialization function
|
IWeightInit |
getWeightInitFunction(Distribution distribution)
Create an instance of the weight initialization function
|
static WeightInit |
valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static WeightInit[] |
values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
public static final WeightInit DISTRIBUTION
public static final WeightInit ZERO
public static final WeightInit ONES
public static final WeightInit SIGMOID_UNIFORM
public static final WeightInit NORMAL
public static final WeightInit LECUN_NORMAL
public static final WeightInit UNIFORM
public static final WeightInit XAVIER
public static final WeightInit XAVIER_UNIFORM
public static final WeightInit XAVIER_FAN_IN
public static final WeightInit XAVIER_LEGACY
public static final WeightInit RELU
public static final WeightInit RELU_UNIFORM
public static final WeightInit IDENTITY
public static final WeightInit LECUN_UNIFORM
public static final WeightInit VAR_SCALING_NORMAL_FAN_IN
public static final WeightInit VAR_SCALING_NORMAL_FAN_OUT
public static final WeightInit VAR_SCALING_NORMAL_FAN_AVG
public static final WeightInit VAR_SCALING_UNIFORM_FAN_IN
public static final WeightInit VAR_SCALING_UNIFORM_FAN_OUT
public static final WeightInit VAR_SCALING_UNIFORM_FAN_AVG
public static WeightInit[] values()
for (WeightInit c : WeightInit.values()) System.out.println(c);
public static WeightInit valueOf(String name)
name
- the name of the enum constant to be returned.IllegalArgumentException
- if this enum type has no constant with the specified nameNullPointerException
- if the argument is nullpublic IWeightInit getWeightInitFunction()
IWeightInit
instancepublic IWeightInit getWeightInitFunction(Distribution distribution)
distribution
- Distribution of the weights (Only used in case DISTRIBUTION)IWeightInit
instanceCopyright © 2022. All rights reserved.