TensorBuffer

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
createDynamic(dataType DataType)
Crea un TensorBuffer dinámico vacío con el DataType especificado.
estático TensorBuffer
createFixedSize(int[] shape, DataType dataType)
Crea un TensorBuffer con shape y DataType especificados.
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
loadBuffer(búfer ByteBuffer)
Carga un búfer de bytes en este TensorBuffer.
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á.

público estático TensorBuffer . createFixedSize (int[] shape, DataType dataType)

Crea un