public LSTM(int units, int inputDim, ActivationLambda activation, ActivationLambda recurrentActivation) { this.units = units; this.inputDim = inputDim; this.activation = activation; this.recurrentActivation = recurrentActivation; }
public SimpleRNNLayer(int units, int inputDim, ActivationLambda lambda) { this.units = units; this.inputDim = inputDim; this.lambda = lambda; prevOutput = new Data2D(1, 1, units, 1); h = new Data2D(1, 1, units, 1); }
public GRULayer(int units, int inputDim, ActivationLambda activation, ActivationLambda recurrentActivation) { this.units = units; this.inputDim = inputDim; this.activation = activation; this.recurrentActivation = recurrentActivation; z = new Data2D(1, 1, units, 1); r = new Data2D(1, 1, units, 1); hh = new Data2D(1, 1, units, 1); h = new Data2D(1, 1, units, 1); }
public LSTMLayer(int units, int inputDim, ActivationLambda activation, ActivationLambda recurrentActivation) { this.units = units; this.inputDim = inputDim; this.activation = activation; this.recurrentActivation = recurrentActivation; h = new Data2D(1, 1, units, 1); C = new Data2D(1, 1, units, 1); ou = new Data2D(1, 1, units, 1); C0 = new Data2D(1, 1, units, 1); i = new Data2D(1, 1, units, 1); f = new Data2D(1, 1, units, 1); }
public SimpleRNN(int units, int inputDim, ActivationLambda lambda) { this.units = units; this.inputDim = inputDim; this.lambda = lambda; }