Посетите симпозиум «Женщины в машинном обучении» 7 декабря Зарегистрируйтесь сейчас

Tensor

открытый интерфейс Тензор

Типизированный многомерный массив, используемый в Tensorflow Lite.

Уроженец ручка Tensor управляется NativeInterpreterWrapper , и не нужно было быть закрыто клиентом. Однако, как только NativeInterpreterWrapper был закрыт, тензор ручка будет недействительна.

Вложенные классы

класс Tensor.QuantizationParams Параметры квантования , что соответствует таблице, QuantizationParameters , в файле схемы TFLite модели.

Публичные методы

абстрактный ByteBuffer
asReadOnlyBuffer ()
Возвращает только для чтения ByteBuffer просмотра данных тензорных.
абстрактный DataType
Datatype ()
Возвращает DataType элементов , хранящихся в тензоре.
абстрактный int
numBytes ()
Возвращает размер тензорных данных в байтах.
абстрактный int
numDimensions ()
Возвращает количество измерений (иногда называют рангом ) тензора.
абстрактный int
numElements ()
Возвращает количество элементов в плоском (одномерном) виде тензора.
абстрактные Tensor.QuantizationParams
quantizationParams ()
Возвращает параметры квантования тензора в владеющем Interpreter .
абстрактный int []
Форма ()
Возвращает форму тензорной, то есть, размеры каждого измерения.
абстрактный int []
shapeSignature ()
Возвращает исходную форму тензорной, то есть, размеры каждого измерения - до того , как изменение размера было выполнено.

Публичные методы

общественный абстрактный ByteBuffer asReadOnlyBuffer ()

Возвращает только для чтения ByteBuffer просмотра данных тензорных.

В общем, этот метод наиболее полезен для получения только для чтения , вида выходного тензора данных, * * после вывода было выполнено (например, с помощью Interpreter.run(Object, Object) ). В частности, некоторые графы динамически в форме выходов, которые могут сделать подачу предварительно определенного выходного буфера в Interpreter неудобно. Пример использования:

interpreter.run(input, null);
 ByteBuffer outputBuffer = interpreter.getOutputTensor(0).asReadOnlyBuffer();
 // Copy or read from outputBuffer.
 

ПРЕДУПРЕЖДЕНИЕ: если тензор еще не назначен, например, до выполнения вывода, результат не определен. Обратите внимание, что базовый указатель тензора может также измениться, когда тензор недействителен каким-либо образом (например, если выполняется вывод или изменяется размер графика), поэтому * небезопасно * хранить ссылку на возвращенный буфер за пределами немедленного использования. непосредственно после вывода. Пример * плохой * использование:

ByteBuffer outputBuffer = interpreter.getOutputTensor(0).asReadOnlyBuffer();
 interpreter.run(input, null);
 // Copy or read from outputBuffer (which may now be invalid).
 

Броски
IllegalArgumentException если тензорные данные не были размещены.

общественный абстрактный DataType DATATYPE ()

Возвращает DataType элементов , хранящихся в тензоре.

общественный абстрактный ИНТ numBytes ()

Возвращает размер тензорных данных в байтах.

общественный абстрактный ИНТ numDimensions ()

Возвращает количество измерений (иногда называют рангом ) тензора.

Будет 0 для скаляра, 1 для вектора, 2 для матрицы, 3 для 3-мерного тензора и т. Д.

общественный абстрактный ИНТ numElements ()

Возвращает количество элементов в плоском (одномерном) виде тензора.

общественный абстрактный Tensor.QuantizationParams quantizationParams ()

Возвращает параметры квантования тензора в владеющем Interpreter .

Только квантованных тензоры имеют действительные QuantizationParameters . Для тензора, который не квантован, значения scale и zero_point равны 0.

общественный абстрактный ИНТ [] форма ()

Возвращает форму тензорной, то есть, размеры каждого измерения.

Возврат
  • массив, где i-й элемент - это размер i-го измерения тензора.

общественный абстрактный ИНТ [] shapeSignature ()

Возвращает исходную форму тензорной, то есть, размеры каждого измерения - до того , как изменение размера было выполнено. Неизвестные размеры обозначаются значением -1.

Возврат
  • массив, где i-й элемент - это размер i-го измерения тензора.