OnnxTensor.fromInputIds constructor

OnnxTensor.fromInputIds(
  1. List<int> inputIds
)

Creates a tensor for BERT-style input IDs.

Implementation

factory OnnxTensor.fromInputIds(List<int> inputIds) {
  final data = calloc<Float>(inputIds.length);
  for (var i = 0; i < inputIds.length; i++) {
    data[i] = inputIds[i].toDouble();
  }

  return OnnxTensor._(
    data: data,
    shape: [1, inputIds.length], // Batch size 1
    elementCount: inputIds.length,
  );
}