Package | Description |
---|---|
org.deeplearning4j.nn.conf | |
org.deeplearning4j.nn.conf.layers | |
org.deeplearning4j.nn.weights |
Modifier and Type | Field and Description |
---|---|
protected WeightInit |
NeuralNetConfiguration.Builder.weightInit |
Modifier and Type | Method and Description |
---|---|
NeuralNetConfiguration.Builder |
NeuralNetConfiguration.Builder.weightInit(WeightInit weightInit)
Weight initialization scheme.
|
Modifier and Type | Field and Description |
---|---|
protected WeightInit |
Layer.weightInit |
protected WeightInit |
Layer.Builder.weightInit |
Modifier and Type | Method and Description |
---|---|
T |
Layer.Builder.weightInit(WeightInit weightInit)
Weight initialization scheme.
|
Modifier and Type | Method and Description |
---|---|
static WeightInit |
WeightInit.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static WeightInit[] |
WeightInit.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Modifier and Type | Method and Description |
---|---|
static org.nd4j.linalg.api.ndarray.INDArray |
WeightInitUtil.initWeights(int[] shape,
WeightInit initScheme,
org.nd4j.linalg.api.rng.distribution.Distribution dist,
char order,
org.nd4j.linalg.api.ndarray.INDArray paramView) |
static org.nd4j.linalg.api.ndarray.INDArray |
WeightInitUtil.initWeights(int[] shape,
WeightInit initScheme,
org.nd4j.linalg.api.rng.distribution.Distribution dist,
org.nd4j.linalg.api.ndarray.INDArray paramView)
Initializes a matrix with the given weight initialization scheme.
|
static org.nd4j.linalg.api.ndarray.INDArray |
WeightInitUtil.initWeights(int nIn,
int nOut,
WeightInit initScheme,
org.nd4j.linalg.api.rng.distribution.Distribution dist,
org.nd4j.linalg.api.ndarray.INDArray paramView)
Initializes a matrix with the given weight initialization scheme
|
Copyright © 2016. All Rights Reserved.