org.nd4j.weightinit

Interface WeightInitScheme

All Known Implementing Classes:

BaseWeightInitScheme, ConstantInitScheme, DistributionInitScheme, IdentityInitScheme, LecunUniformInitScheme, NDArraySupplierInitScheme, OneInitScheme, ReluInitScheme, ReluUniformInitScheme, SigmoidUniformInitScheme, UniformInitScheme, VarScalingNormalFanAvgInitScheme, VarScalingNormalFanInInitScheme, VarScalingNormalFanOutInitScheme, VarScalingNormalUniformFanInInitScheme, VarScalingNormalUniformFanOutInitScheme, VarScalingUniformFanAvgInitScheme, XavierFanInInitScheme, XavierInitScheme, XavierUniformInitScheme, ZeroInitScheme

public interface WeightInitScheme

Defines weight initialization for neural networks. Use BaseWeightInitScheme to create a new WeightInitScheme This is needed to handle things like the parameters view.

Author:

Adam Gibson

Method Summary

Modifier and Type	Method and Description
INDArray	create(int[] shape) Create the array
INDArray	create(int[] shape, INDArray paramsView) Create the array
char	order() The order of the weight init
WeightInit	type() The type of the weight init

Method Detail

create

INDArray create(int[] shape,
                INDArray paramsView)

Create the array

Parameters:

shape - the shape of the array

paramsView - the parameters view

Returns:

the created array

create

INDArray create(int[] shape)

Create the array

Parameters:

shape - the shape of the array

Returns:

the created array

order

char order()

The order of the weight init

Returns:

type

WeightInit type()

The type of the weight init

Returns: