nn_lstm_step: Build one LSTM step

In ggmlR: 'GGML' Tensor Operations for Machine Learning

Build one LSTM step

Description

Build one LSTM step

Usage

nn_lstm_step(
  ctx,
  x_t,
  h_t,
  c_t,
  W_gates,
  U_gates,
  b_gates,
  units,
  act_cell,
  act_rec
)

Arguments

ctx	ggml compute context
x_t	Input at this step: tensor [input_size, N]
h_t	Previous hidden state: tensor [units, N]
c_t	Previous cell state: tensor [units, N]
W_gates	[input_size, 4*units]
U_gates	[units, 4*units]
b_gates	[4*units]
units	Integer
act_cell	Activation name for cell gate (tanh by default)
act_rec	Activation name for i/f/o gates (sigmoid by default)

Value

list(h = new_h, c = new_c)

ggmlR documentation built on July 14, 2026, 1:08 a.m.