ggml_opt_dataset_get_batch_head: Get one output head's slice of a batch's labels

In ggmlR: 'GGML' Tensor Operations for Machine Learning

Get one output head's slice of a batch's labels

Description

Multi-output counterpart of ggml_opt_dataset_get_batch(). The dataset's labels hold every head's labels concatenated along the first dimension; this copies just the slice belonging to one head.

Usage

ggml_opt_dataset_get_batch_head(
  dataset,
  data_batch,
  labels_batch,
  labels_off,
  ibatch
)

Arguments

dataset	Dataset pointer
data_batch	Input batch tensor, or NULL to copy only the labels (the caller copies the inputs once, with the first head)
labels_batch	This head's labels batch tensor
labels_off	Offset of this head within a label row, 0-based, in elements
ibatch	Batch index, 0-based

Value

NULL, invisibly

See Also

Other optimization: ggml_fit_opt(), ggml_fit_opt_gen(), ggml_fit_opt_multi(), ggml_opt_alloc(), ggml_opt_context_optimizer_type(), ggml_opt_dataset_data(), ggml_opt_dataset_free(), ggml_opt_dataset_get_batch(), ggml_opt_dataset_init(), ggml_opt_dataset_labels(), ggml_opt_dataset_ndata(), ggml_opt_dataset_shuffle(), ggml_opt_dataset_weights(), ggml_opt_default_params(), ggml_opt_epoch(), ggml_opt_eval(), ggml_opt_fit(), ggml_opt_fit_multi(), ggml_opt_free(), ggml_opt_get_lr(), ggml_opt_grad_acc(), ggml_opt_init(), ggml_opt_init_for_fit(), ggml_opt_init_for_fit_multi(), ggml_opt_inputs(), ggml_opt_labels(), ggml_opt_loss(), ggml_opt_loss_i(), ggml_opt_loss_type_binary_cross_entropy(), ggml_opt_loss_type_cross_entropy(), ggml_opt_loss_type_huber(), ggml_opt_loss_type_mae(), ggml_opt_loss_type_mean(), ggml_opt_loss_type_mse(), ggml_opt_loss_type_sum(), ggml_opt_loss_type_weighted_mse(), ggml_opt_n_loss(), ggml_opt_ncorrect(), ggml_opt_optimizer_name(), ggml_opt_optimizer_type_adamw(), ggml_opt_optimizer_type_sgd(), ggml_opt_outputs(), ggml_opt_outputs_i(), ggml_opt_pred(), ggml_opt_prepare_alloc(), ggml_opt_reset(), ggml_opt_result_accuracy(), ggml_opt_result_free(), ggml_opt_result_init(), ggml_opt_result_loss(), ggml_opt_result_loss_i(), ggml_opt_result_n_loss(), ggml_opt_result_ndata(), ggml_opt_result_pred(), ggml_opt_result_reset(), ggml_opt_set_lr(), ggml_opt_static_graphs(), ggml_snapshot_free(), ggml_trainer()

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