getOutputTensor method

Tensor getOutputTensor(
  1. int index
)

Implementation

Tensor getOutputTensor(int index) {
  final int tensorsCount = TfLiteInterpreterGetOutputTensorCount(_ref);
  if (index < 0 || index >= tensorsCount) {
    throw ArgumentError('Invalid output Tensor index: $index');
  }
  return Tensor(TfLiteInterpreterGetOutputTensor(_ref, index));
}