View source: R/nn_functional.R
| nn_build_functional_graph | R Documentation |
Build ggml computation graph for a functional model
nn_build_functional_graph(model, batch_size, training = FALSE)
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). |
Named list with inputs, outputs, ctx_weights, ctx_compute, buffer, node_weights
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.