TensorBuffer

classe astratta pubblica TensorBuffer
Sottoclassi dirette note

Rappresenta il buffer dei dati per l'input o l'output di un modello.

Metodi pubblici

statico TensorBuffer
createDynamic(DataType dataType)
Crea un elemento TensorBuffer dinamico vuoto con il valore DataType specificato.
statico TensorBuffer
createFixedSize(int[] shape, DataType dataType)
Crea un elemento TensorBuffer con shape e DataType specificati.
statico TensorBuffer
createFrom(buffer TensorBuffer, DataType dataType)
Crea un TensorBuffer di copia approfondita dei dati da un altro, con il valore DataType specificato.
ByteBuffer
getBuffer()
Restituisce il buffer di dati.
astratto DataType
getDataType()
Restituisce il tipo di dati del buffer.
int
getFlatSize()
Ottiene il valore flatSize del buffer.
astratto float[]
getFloatArray()
Restituisce un array in virgola mobile dei valori memorizzati in questo buffer.
astratto in virgola mobile
getFloatValue(int ABIndex)
Restituisce un valore in virgola mobile in un dato indice.
astratto int[]
getIntArray()
Restituisce un array int dei valori archiviati nel buffer.
astratto int
getIntValue(int ABIndex)
Restituisce un valore int in un determinato indice.
int[]
getShape()
Recupera la forma corrente.
astratto int
getTypeSize()
Restituisce il numero di byte di un singolo elemento dell'array.
booleano
isDynamic()
Restituisce un valore che indica se TensorBuffer ha dimensioni dinamiche (il ridimensionamento potrebbe avvenire in modo arbitrario).
astratto null
loadArray(int[] src, int[] shape)
Carica un array int in questo buffer con una forma specifica.
astratto null
loadArray(float[] src, int[] shape)
Carica un array in virgola mobile in questo buffer con una forma specifica.
null
loadArray(float[] src)
Carica un array in virgola mobile in questo buffer.
null
loadArray(int[] src)
Carica un array int in questo buffer.
null
loadBuffer(buffer ByteBuffer)
Carica un buffer di byte in questo TensorBuffer.
null
loadBuffer(buffer ByteBuffer, forma int[])
Carica un buffer di byte in questo TensorBuffer con una forma specifica.

Metodi ereditati

Metodi pubblici

pubblica statico TensorBuffer createDynamic (DataType dataType)

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

Dynamic TensorBuffers rialloca la memoria quando vengono caricati array o buffer di dati dimensioni del buffer diverse. 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.
 

Parametri
dataType Il dataType di TensorBuffer da creare.

pubblica statico TensorBuffer createFixedSize (int[] shape, DataType dataType)

Crea un elemento TensorBuffer con shape e DataType specificati. Ecco alcuni esempi 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);
 

Dopo la creazione, non è possibile modificare la dimensione di un TensorBuffer a dimensione fissa.

Parametri
<
forma La forma di TensorBuffer da creare.
dataType Il dataType di TensorBuffer da creare.