nn_gru_step: Build one GRU step

In ggmlR: 'GGML' Tensor Operations for Machine Learning

Build one GRU step

Description

Build one GRU step

Usage

nn_gru_step(
  ctx,
  x_t,
  h_t,
  W_zh,
  U_zh,
  b_zh,
  W_n,
  U_n,
  b_n,
  units,
  act_cell,
  act_rec
)

Arguments

ctx	ggml compute context
x_t	Input at this step: [input_size, N]
h_t	Previous hidden state: [units, N]
W_zh	[input_size, 2*units]
U_zh	[units, 2*units]
b_zh	[2*units]
W_n	[input_size, units]
U_n	[units, units]
b_n	[units]
units	Integer
act_cell	Activation for candidate (tanh)
act_rec	Activation for z/r gates (sigmoid)

Value

list(h = new_h)

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