Package io.github.kirstenali.deepj.nn

Class LSTMCell

java.lang.Object

io.github.kirstenali.deepj.nn.Module

io.github.kirstenali.deepj.nn.RecurrentCell

io.github.kirstenali.deepj.nn.LSTMCell

All Implemented Interfaces:

Layer, Trainable

public final class LSTMCell
extends RecurrentCell

LSTM with input/forget/candidate/output gate order and explicit cell memory.

Field Summary

Fields inherited from class io.github.kirstenali.deepj.nn.RecurrentCell

hiddenProjection, inputProjection

Constructor Summary

Constructors

Constructor	Description
LSTMCell(int inputs, int hidden, Random random)
LSTMCell(int inputs, int hidden, Random random, ParameterFactory parameters)

Method Summary

Modifier and Type	Method	Description
protected boolean	hasCellMemory()
protected RecurrentState	next(Tensor input, RecurrentState state)

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

children, forward, gate, hiddenSize, initialState, inputSize, step

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

Constructor Details

LSTMCell

public LSTMCell(int inputs,
 int hidden,
 Random random)

LSTMCell

public LSTMCell(int inputs,
 int hidden,
 Random random,
 ParameterFactory parameters)

Method Details

hasCellMemory

protected boolean hasCellMemory()

Overrides:

hasCellMemory in class RecurrentCell

next

protected RecurrentState next(Tensor input,
 RecurrentState state)

Specified by:

next in class RecurrentCell