Package io.github.kirstenali.deepj.nn

Class RecurrentCell

java.lang.Object

io.github.kirstenali.deepj.nn.Module

io.github.kirstenali.deepj.nn.RecurrentCell

All Implemented Interfaces:

Layer, Trainable

Direct Known Subclasses:

GRUCell, LSTMCell, RNNCell

public abstract class RecurrentCell
extends Module

Shared input/state validation and DeepJ linear projections for recurrent cells.

Field Summary

Fields

Modifier and Type	Field	Description
protected final Linear	hiddenProjection
protected final Linear	inputProjection

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	RecurrentCell(int inputs, int hidden, int gates, Random random, ParameterFactory parameters)

Method Summary

Modifier and Type	Method	Description
protected List<Layer>	children()
final Tensor	forward(Tensor input)
protected final Tensor	gate(Tensor projection, int index)
protected boolean	hasCellMemory()
final int	hiddenSize()
final RecurrentState	initialState(int batch, DType dtype)
final int	inputSize()
protected abstract RecurrentState	next(Tensor input, RecurrentState state)
final RecurrentState	step(Tensor input, RecurrentState state)

Methods inherited from class io.github.kirstenali.deepj.nn.Module

buffers, eval, isTraining, parameters, setTraining, train

Methods inherited from class java.lang.Object

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

Methods inherited from interface io.github.kirstenali.deepj.layers.Layer

forward, supportsIncrementalInference

Methods inherited from interface io.github.kirstenali.deepj.training.Trainable

zeroGrad

Field Details

inputProjection

protected final Linear inputProjection

hiddenProjection

protected final Linear hiddenProjection

Constructor Details

RecurrentCell

protected RecurrentCell(int inputs,
 int hidden,
 int gates,
 Random random,
 ParameterFactory parameters)

Method Details

inputSize

public final int inputSize()

hiddenSize

public final int hiddenSize()

hasCellMemory

protected boolean hasCellMemory()

next

protected abstract RecurrentState next(Tensor input,
 RecurrentState state)

children

protected List<Layer> children()

Overrides:

children in class Module

initialState

public final RecurrentState initialState(int batch,
 DType dtype)

step

public final RecurrentState step(Tensor input,
 RecurrentState state)

forward

public final Tensor forward(Tensor input)

gate

protected final Tensor gate(Tensor projection,
 int index)