LSTMBlockCell

public final class LSTMBlockCell

Calcula a propagação direta da célula LSTM para um intervalo de tempo.

Esta implementação usa 1 matriz de peso e 1 vetor de polarização, e há uma conexão de olho mágico opcional.

Esta op do kernel implementa as seguintes equações matemáticas:

xh = [x, h_prev]
 [i, f, ci, o] = xh * w + b
 f = f + forget_bias
 
 if not use_peephole:
   wci = wcf = wco = 0
 
 i = sigmoid(cs_prev * wci + i)
 f = sigmoid(cs_prev * wcf + f)
 ci = tanh(ci)
 
 cs = ci .* i + cs_prev .* f
 cs = clip(cs, cell_clip)
 
 o = sigmoid(cs * wco + o)
 co = tanh(cs)
 h = co .* o
 

Classes aninhadas

classe LSTMBlockCell.Options Atributos opcionais para LSTMBlockCell

Métodos Públicos

estáticos LSTMBlockCell.Options
cellClip (Flutuador cellClip)
Output <T>
ci ()
A entrada da célula.
Output <T>
co ()
A célula depois do tanh.
estática <T estende Number> LSTMBlockCell <T>
criar ( Scope escopo, Operando <T> x, Operando <T> csPrev, Operando <T> hPrev, Operando <T> w, Operando <T> wci, Operando <T> WCF, Operando <T> OMA, Operando <T > b, Options ... opções)
Método de fábrica para criar uma classe envolvendo uma nova operação LSTMBlockCell.
Output <T>
cs ()
O estado da célula antes do tanh.
Output <T>
f ()
O portão de esquecimento.
estáticos LSTMBlockCell.Options
forgetBias (float) forgetBias
Output <T>
h ()
O vetor h de saída.
Output <T>
i ()
A porta de entrada.
Output <T>
O ()
O portão de saída.
estáticos LSTMBlockCell.Options
usePeephole (booleano usePeephole)

Métodos herdados

Métodos Públicos

public static LSTMBlockCell.Options cellClip (Float cellClip)

Parâmetros
cellClip Valor para o qual cortar o valor 'cs'.

pública Output <T> ci ()

A entrada da célula.

pública Output <T> co ()

A célula depois do tanh.

public static LSTMBlockCell <T> create ( Scope escopo, Operando <T> x, Operando <T> csPrev, Operando <T> hPrev, Operando <T> w, Operando <T> wci, Operando <T> WCF, Operando <T > OMA, Operando <T> b, Options ... opções)

Método de fábrica para criar uma classe envolvendo uma nova operação LSTMBlockCell.

Parâmetros
alcance escopo atual
x A entrada para a célula LSTM, forma (batch_size, num_inputs).
csPrev Valor do estado da célula na etapa de tempo anterior.
hPrev Saída da célula anterior na etapa de tempo anterior.
C A matriz de peso.
wci A matriz de peso para a conexão do olho mágico da porta de entrada.
wcf A matriz de peso para esquecer a conexão do olho mágico do portão.
wco A matriz de peso para a conexão do olho mágico do portão de saída.
b O vetor de polarização.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de LSTMBlockCell

pública Output <T> cs ()

O estado da célula antes do tanh.

pública Output <T> f ()

O portão de esquecimento.

public static LSTMBlockCell.Options forgetBias (forgetBias float)

Parâmetros
esquecerBias A tendência para esquecer o portão.

pública Output <T> h ()

O vetor h de saída.

pública Output <T> i ()

A porta de entrada.

pública Output <T> o ()

O portão de saída.

public static LSTMBlockCell.Options usePeephole (Boolean usePeephole)

Parâmetros
usePeephole Se usar pesos de olho mágico.