clase abstracta pública
TensorBuffer
Subclases directas conocidas |
Representa el búfer de datos para la entrada o la salida de un modelo.
Métodos públicos
estático TensorBuffer | |
estático TensorBuffer |
createFixedSize(int[] shape, DataType dataType)
|
estático TensorBuffer |
createFrom(búfer TensorBuffer, dataType de DataType)
Crea un
TensorBuffer de copia profunda de datos de otro, con el DataType especificado. |
ByteBuffer |
getBuffer()
Muestra el búfer de datos.
|
abstracto DataType |
getDataType()
Muestra el tipo de datos de este búfer.
|
int |
getFlatSize()
Obtiene el plano del tamaño del búfer.
|
abstracto float[] |
getFloatArray()
Muestra un array de número de punto flotante de los valores almacenados en este búfer.
|
abstracto número de punto flotante |
getFloatValue(int absIndex)
Muestra un valor flotante en un índice determinado.
|
abstracto número entero |
getIntArray()
Muestra un array int de los valores almacenados en este búfer.
|
abstracto número entero |
getIntValue(int absIndex)
Muestra un valor int en un índice determinado.
|
int[] |
getShape()
Obtiene la forma actual.
|
abstracto número entero |
getTypeSize()
Muestra la cantidad de bytes de un solo elemento del array.
|
boolean |
isDynamic()
Se muestra si
TensorBuffer tiene un tamaño dinámico (puede hacerlo de manera arbitraria). |
abstracto anular |
loadArray(int[] src, int[] shape)
Carga un array de int en este búfer con una forma específica.
|
abstracto anular |
loadArray(float[] src, int[] shape)
Carga un array de número de punto flotante en este búfer con una forma específica.
|
void |
loadArray(float[] src)
Carga un array de número de punto flotante en este búfer.
|
void |
loadArray(int[] src)
Carga un array de int en este búfer.
|
void | |
void |
loadBuffer(búfer ByteBuffer, forma int[])
Carga un búfer de bytes en este
TensorBuffer con una forma específica. |
Métodos heredados
Métodos públicos
público estático TensorBuffer . createDynamic (DataType)
Crea un TensorBuffer
dinámico vacío con el DataType
especificado. La forma del
creado el TensorBuffer
es {0}.
Los TensorBuffers dinámicos reasignarán memoria cuando se carguen arrays o búferes de datos de diferentes tamaños de búfer. Estos son algunos ejemplos:
// 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.
Parámetros
dataType | El dataType del TensorBuffer que se creará.
|
---|