TensorBuffer

Crea un TensorBuffer dinamico vuoto con DataType specificato. La forma del TensorBuffer creato è {0}.

Dynamic TensorBuffers riallocherà la memoria durante il caricamento di array o buffer di dati di diverse dimensioni di buffer. Ecco alcuni esempi:

 // Creating a float dynamic TensorBuffer:
 TensorBuffer tensorBuffer = TensorBuffer.createDynamic(DataType.FLOAT32);
 // Loading a float array:
 float[] arr1 = new float[] {1, 2, 3};
 tensorBuffer.loadArray(arr, new int[] {arr1.length});
 // loading another float array:
 float[] arr2 = new float[] {1, 2, 3, 4, 5};
 tensorBuffer.loadArray(arr, new int[] {arr2.length});
 // loading a third float array with the same size as arr2, assuming shape doesn't change:
 float[] arr3 = new float[] {5, 4, 3, 2, 1};
 tensorBuffer.loadArray(arr);
 // loading a forth float array with different size as arr3 and omitting the shape will result
 // in error:
 float[] arr4 = new float[] {3, 2, 1};
 tensorBuffer.loadArray(arr); // Error: The size of byte buffer and the shape do not match.
 

Crea un TensorBuffer con shape e DataType specificati. Ecco alcuni esempi:

 // Creating a float TensorBuffer with shape {2, 3}:
 int[] shape = new int[] {2, 3};
 TensorBuffer tensorBuffer = TensorBuffer.createFixedSize(shape, DataType.FLOAT32);
 

 // Creating an uint8 TensorBuffer of a scalar:
 int[] shape = new int[] {};
 TensorBuffer tensorBuffer = TensorBuffer.createFixedSize(shape, DataType.UINT8);
 

 // Creating an empty uint8 TensorBuffer:
 int[] shape = new int[] {0};
 TensorBuffer tensorBuffer = TensorBuffer.createFixedSize(shape, DataType.UINT8);
 

La dimensione di un TensorBuffer a dimensione fissa non può essere modificata una volta creato.

Crea un TensorBuffer che copia in profondità i dati da un altro, con DataType specificato.

Restituisce il buffer dei dati.

Restituisce il tipo di dati di questo buffer.

Ottiene flatSize del buffer.

Restituisce un array float dei valori memorizzati in questo buffer. Se il buffer è di tipo diverso da float, i valori verranno convertiti in float. Ad esempio, i valori in TensorBufferUint8 verranno convertiti da uint8 a float.

Restituisce un valore float in corrispondenza di un determinato indice. Se il buffer è di tipo diverso da float, il valore verrà convertito in float. Ad esempio, quando si legge un valore da TensorBufferUint8 , il valore verrà prima letto come uint8 e quindi verrà convertito da uint8 a float.

 For example, a TensorBuffer with shape {2, 3} that represents the following array,
 [[0.0f, 1.0f, 2.0f], [3.0f, 4.0f, 5.0f]].

 The fourth element (whose value is 3.0f) in the TensorBuffer can be retrieved by:
 float v = tensorBuffer.getFloatValue(3);
 

Restituisce un array int dei valori memorizzati in questo buffer. Se il buffer è di tipo diverso da int, i valori verranno convertiti in int e potrebbe verificarsi una perdita di precisione. Ad esempio, ottenendo un array int da TensorBufferFloat con valori {400.32f, 23.04f}, l'output è {400, 23}.

Restituisce un valore intero in corrispondenza di un determinato indice. Se il buffer è di tipo diverso da int, il valore verrà convertito in int. Ad esempio, quando si legge un valore da TensorBufferFloat , il valore verrà prima letto come float e quindi verrà convertito da float a int. Potrebbe verificarsi una perdita di precisione.

 For example, a TensorBuffer with shape {2, 3} that represents the following array,
 [[0.0f, 1.0f, 2.0f], [3.0f, 4.0f, 5.0f]].

 The fourth element (whose value is 3.0f) in the TensorBuffer can be retrieved by:
 int v = tensorBuffer.getIntValue(3);
 Note that v is converted from 3.0f to 3 as a result of type conversion.
 

Ottiene la forma corrente. (restituendo una copia qui per evitare modifiche impreviste.)

Restituisce il numero di byte di un singolo elemento nell'array. Ad esempio, un buffer float restituirà 4 e un buffer byte restituirà 1.

Restituisce se TensorBuffer ha dimensioni dinamiche (potrebbe ridimensionarsi arbitrariamente).

Carica un array int in questo buffer con una forma specifica. Se il buffer è di tipo diverso da int, i valori verranno convertiti nel tipo del buffer prima di essere caricati nel buffer e potrebbe verificarsi una perdita di precisione. Ad esempio, caricando un array int con valori {400, -23} in un TensorBufferUint8 , i valori verranno bloccati su [0, 255] e quindi convertiti in uint8 da {255, 0}.

Carica un array float in questo buffer con una forma specifica. Se il buffer è di tipo diverso da float, i valori verranno convertiti nel tipo del buffer prima di essere caricati nel buffer e potrebbe verificarsi una perdita di precisione. Ad esempio, caricando un array float in un TensorBufferUint8 con valori {400.32f, -23.04f}, i valori verranno bloccati su [0, 255] e quindi convertiti in uint8 da {255, 0}.

Carica un array float in questo buffer. Se il buffer è di tipo diverso da float, i valori verranno convertiti nel tipo del buffer prima di essere caricati nel buffer e potrebbe verificarsi una perdita di precisione. Ad esempio, caricando un array float in un TensorBufferUint8 con valori {400.32f, -23.04f}, i valori verranno bloccati su [0, 255] e quindi convertiti in uint8 da {255, 0}.

L'utilizzo di questo metodo presuppone che la forma di src sia la stessa della forma di questo TensorBuffer . Pertanto, la dimensione del buffer ( src.length ) dovrebbe sempre corrispondere alla dimensione piatta di questo TensorBuffer , sia per TensorBuffer a dimensione fissa che dinamico. Utilizzare loadArray(float[], int[]) se src ha una forma diversa.

Carica un array int in questo buffer. Se il buffer è di tipo diverso da int, i valori verranno convertiti nel tipo del buffer prima di essere caricati nel buffer e potrebbe verificarsi una perdita di precisione. Ad esempio, caricando un array int con valori {400, -23} in un TensorBufferUint8 , i valori verranno bloccati su [0, 255] e quindi convertiti in uint8 da {255, 0}.

L'utilizzo di questo metodo presuppone che la forma di src sia la stessa della forma di questo TensorBuffer . Pertanto, la dimensione del buffer ( src.length ) dovrebbe sempre corrispondere alla dimensione piatta di questo TensorBuffer , sia per TensorBuffer a dimensione fissa che dinamico. Utilizzare loadArray(int[], int[]) se src ha una forma diversa.

Carica un buffer di byte in questo TensorBuffer . La dimensione del buffer deve corrispondere alla dimensione flat di questo TensorBuffer .

L'utilizzo di questo metodo presuppone che la forma del buffer sia la stessa della forma di questo TensorBuffer . Pertanto, la dimensione del buffer ( buffer.limit() ) dovrebbe sempre corrispondere alla dimensione piatta di questo TensorBuffer , sia per TensorBuffer a dimensione fissa che dinamico. Utilizzare loadBuffer(ByteBuffer, int[]) se buffer ha una forma diversa.

Importante: il buffer caricato è un riferimento. NON MODIFICARE. Non creiamo una copia qui per motivi di prestazioni, ma se è necessaria una modifica, creane una copia.

Per ottenere prestazioni ottimali, caricare sempre un ByteBuffer diretto o un ByteBuffer supportato da un array.

Se il buffer è di sola lettura, adottiamo una strategia copy-on-write per migliorare le prestazioni.

Carica un buffer di byte in questo TensorBuffer con una forma specifica.

Importante: il buffer caricato è un riferimento. NON MODIFICARE. Non creiamo una copia qui per motivi di prestazioni, ma se è necessaria una modifica, creane una copia.

Per ottenere prestazioni ottimali, caricare sempre un ByteBuffer diretto o un ByteBuffer supportato da un array.