Tensor

공개 인터페이스 Tensor

Tensorflow Lite에서 사용되는 형식화된 다차원 배열입니다.

Tensor 의 기본 핸들은 NativeInterpreterWrapper 에 의해 관리되며 클라이언트가 닫을 필요가 없습니다. 그러나 NativeInterpreterWrapper 가 닫히면 텐서 핸들이 무효화됩니다.

중첩 클래스

수업 Tensor.QuantizationParams TFLite 모델 스키마 파일의 QuantizationParameters 테이블에 해당하는 양자화 매개변수입니다 .

공개 방법

추상 바이트버퍼
asReadOnlyBuffer ()
텐서 데이터의 읽기 전용 ByteBuffer 뷰를 반환합니다.
추상 데이터 유형
데이터 형식 ()
Tensor에 저장된 요소의 DataType 반환합니다.
추상 정수
숫자바이트 ()
텐서 데이터의 크기를 바이트 단위로 반환합니다.
추상 정수
numDimensions ()
Tensor의 차원 수( 순위 라고도 함)를 반환합니다.
추상 정수
요소 수 ()
텐서의 평면화된(1-D) 뷰에 있는 요소 수를 반환합니다.
추상 Tensor.QuantizationParams
퀀트화파람 ()
소유 인터프리터 내 텐서의 양자화 매개변수를 반환합니다.
추상 정수[]
모양 ()
Tensor의 모양 , 즉 각 차원의 크기를 반환합니다.
추상 정수[]
셰이프시그니처 ()
크기 조정이 수행되기 전의 Tensor의 원래 모양 , 즉 각 차원의 크기를 반환합니다.

공개 방법

공개 추상 ByteBuffer asReadOnlyBuffer ()

텐서 데이터의 읽기 전용 ByteBuffer 뷰를 반환합니다.

일반적으로 이 방법은 추론이 실행된 *후*(예: InterpreterApi.run(Object, Object) 를 통해) 출력 텐서 데이터의 읽기 전용 보기를 얻는 데 가장 유용합니다. 특히 일부 그래프에는 동적으로 모양이 지정된 출력이 있으므로 사전 정의된 출력 버퍼를 인터프리터에 공급하는 것이 어색해질 수 있습니다. 사용 예:

 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 텐서 데이터가 할당되지 않은 경우.

공개 추상 데이터 유형 데이터 유형 ()

Tensor에 저장된 요소의 DataType 반환합니다.

공개 추상 int numBytes ()

텐서 데이터의 크기를 바이트 단위로 반환합니다.

공개 추상 int numDimensions ()

Tensor의 차원 수( 순위 라고도 함)를 반환합니다.

스칼라의 경우 0, 벡터의 경우 1, 행렬의 경우 2, 3차원 텐서의 경우 3입니다.

공개 추상 int numElements ()

텐서의 평면화된(1-D) 뷰에 있는 요소 수를 반환합니다.

공개 추상 Tensor.QuantizationParams yangizationParams ()

소유 인터프리터 내 텐서의 양자화 매개변수를 반환합니다.

양자화된 텐서만 유효한 QuantizationParameters 갖습니다. 양자화되지 않은 텐서의 경우 scale과 zero_point 값은 모두 0입니다.

공개 추상 int[] 모양 ()

Tensor의 모양 , 즉 각 차원의 크기를 반환합니다.

보고
  • i번째 요소가 텐서의 i번째 차원 크기인 배열입니다.

공개 추상 int[] 모양 서명 ()

크기 조정이 수행되기 전의 Tensor의 원래 모양 , 즉 각 차원의 크기를 반환합니다. 알 수 없는 차원은 -1 값으로 지정됩니다.

보고
  • i번째 요소가 텐서의 i번째 차원 크기인 배열입니다.