Learn R Programming

ggmlR (version 0.8.1)

nn_build_lstm: Build LSTM forward pass for Sequential model

Description

Build LSTM forward pass for Sequential model

Usage

nn_build_lstm(ctx, input_tensor, layer, batch_size)

Arguments

Value

A ggml_tensor: last hidden state [units, N], or all hidden states [units, seq_len, N] if return_sequences = TRUE.