nn_build_functional_graph: Build ggml computation graph for a functional model

View source: R/nn_functional.R

nn_build_functional_graphR Documentation

Build ggml computation graph for a functional model

Description

Build ggml computation graph for a functional model

Usage

nn_build_functional_graph(
  model,
  batch_size,
  training = FALSE,
  logits_output = FALSE
)

Arguments

model

A ggml_functional_model

batch_size

Integer batch size

training

Logical; TRUE during fit (activates dropout scaling), FALSE during evaluate/predict (dropout becomes identity).

Value

Named list with inputs, outputs, ctx_weights, ctx_compute, buffer, node_weights


ggmlR documentation built on Oct. 23, 2026, 1:07 a.m.