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)
|
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 | |
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. |
Lanci
NullPointerException | se shape è null. |
---|---|
IllegalArgumentException | se shape contiene elementi non positivi.
|