Package org.DeepJ.ann.layers

Class DenseLayer

java.lang.Object

org.DeepJ.ann.layers.DenseLayer

All Implemented Interfaces:

Layer

public class DenseLayer
extends Object
implements Layer

Constructor Summary

Constructors

Constructor	Description
DenseLayer(int inSize, int outSize)
DenseLayer(int inSize, int outSize, OptimizerFactory factory)

Method Summary

Modifier and Type	Method	Description
Tensor	backward(Tensor gradOut, double learningRate)
Tensor	forward(Tensor in)
Tensor	getBiases()
Tensor	getGradB()
Tensor	getGradW()
Tensor	getWeights()
void	step()

Methods inherited from class java.lang.Object

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

Constructor Details

DenseLayer

public DenseLayer(int inSize,
 int outSize)

DenseLayer

public DenseLayer(int inSize,
 int outSize,
 OptimizerFactory factory)

Method Details

forward

public Tensor forward(Tensor in)

Specified by:

forward in interface Layer

backward

public Tensor backward(Tensor gradOut,
 double learningRate)

Specified by:

backward in interface Layer

step

public void step()

Specified by:

step in interface Layer

getWeights

public Tensor getWeights()

getBiases

public Tensor getBiases()

getGradW

public Tensor getGradW()

getGradB

public Tensor getGradB()