TensorBuffer

Cria um TensorBuffer dinâmico vazio com DataType especificado. A forma do TensorBuffer criado é {0}.

TensorBuffers dinâmicos realocarão memória ao carregar matrizes ou buffers de dados de diferentes tamanhos de buffer. Aqui estão alguns exemplos:

 // 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.
 

Cria um TensorBuffer com shape e DataType especificados. Aqui estão alguns exemplos:

 // 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);
 

O tamanho de um TensorBuffer de tamanho fixo não pode ser alterado depois de criado.

Cria um TensorBuffer copiando dados em profundidade de outro, com DataType especificado.

Retorna o buffer de dados.

Retorna o tipo de dados deste buffer.

Obtém o flatSize do buffer.

Retorna uma matriz flutuante dos valores armazenados neste buffer. Se o buffer for de tipos diferentes de float, os valores serão convertidos em float. Por exemplo, os valores em TensorBufferUint8 serão convertidos de uint8 para float.

Retorna um valor flutuante em um determinado índice. Se o buffer for de tipo diferente de float, o valor será convertido em float. Por exemplo, ao ler um valor de TensorBufferUint8 , o valor será primeiro lido como uint8 e depois será convertido de uint8 em 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);
 

Retorna uma matriz interna dos valores armazenados neste buffer. Se o buffer for de um tipo diferente de int, os valores serão convertidos em int e poderá ocorrer perda de precisão. Por exemplo, obtendo uma matriz int de um TensorBufferFloat com valores {400.32f, 23.04f}, a saída é {400, 23}.

Retorna um valor int em um determinado índice. Se o buffer for de tipo diferente de int, o valor será convertido em int. Por exemplo, ao ler um valor de TensorBufferFloat , o valor será primeiro lido como float e depois convertido de float para int. Pode haver perda de precisão.

 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.
 

Obtém a forma atual. (devolvendo uma cópia aqui para evitar modificações inesperadas.)

Retorna o número de bytes de um único elemento da matriz. Por exemplo, um buffer flutuante retornará 4 e um buffer de bytes retornará 1.

Retorna se o TensorBuffer tiver tamanho dinâmico (pode ser redimensionado arbitrariamente).

Carrega uma matriz int neste buffer com formato específico. Se o buffer for de tipos diferentes de int, os valores serão convertidos no tipo do buffer antes de serem carregados no buffer e poderá ocorrer perda de precisão. Por exemplo, carregando uma matriz int com valores {400, -23} em um TensorBufferUint8 , os valores serão fixados em [0, 255] e então convertidos em uint8 por {255, 0}.

Carrega uma matriz flutuante neste buffer com formato específico. Se o buffer for de tipos diferentes de float, os valores serão convertidos no tipo do buffer antes de serem carregados no buffer e poderá ocorrer perda de precisão. Por exemplo, ao carregar uma matriz flutuante em um TensorBufferUint8 com valores {400.32f, -23.04f}, os valores serão fixados em [0, 255] e então convertidos em uint8 por {255, 0}.

Carrega uma matriz flutuante neste buffer. Se o buffer for de tipos diferentes de float, os valores serão convertidos no tipo do buffer antes de serem carregados no buffer e poderá ocorrer perda de precisão. Por exemplo, ao carregar uma matriz flutuante em um TensorBufferUint8 com valores {400.32f, -23.04f}, os valores serão fixados em [0, 255] e então convertidos em uint8 por {255, 0}.

O uso deste método pressupõe que a forma de src é igual à forma deste TensorBuffer . Portanto, o tamanho do buffer ( src.length ) deve sempre corresponder ao tamanho plano deste TensorBuffer , tanto para TensorBuffer de tamanho fixo quanto para dinâmico. Use loadArray(float[], int[]) se src tiver uma forma diferente.

Carrega uma matriz int neste buffer. Se o buffer for de tipos diferentes de int, os valores serão convertidos no tipo do buffer antes de serem carregados no buffer e poderá ocorrer perda de precisão. Por exemplo, carregando uma matriz int com valores {400, -23} em um TensorBufferUint8 , os valores serão fixados em [0, 255] e então convertidos em uint8 por {255, 0}.

O uso deste método pressupõe que a forma de src é igual à forma deste TensorBuffer . Portanto, o tamanho do buffer ( src.length ) deve sempre corresponder ao tamanho plano deste TensorBuffer , tanto para TensorBuffer de tamanho fixo quanto para dinâmico. Use loadArray(int[], int[]) se src tiver uma forma diferente.

Carrega um buffer de bytes neste TensorBuffer . O tamanho do buffer deve corresponder ao tamanho plano deste TensorBuffer .

O uso deste método pressupõe que o formato do buffer é igual ao formato deste TensorBuffer . Portanto, o tamanho do buffer ( buffer.limit() ) deve sempre corresponder ao tamanho plano deste TensorBuffer , tanto para TensorBuffer de tamanho fixo quanto para dinâmico. Use loadBuffer(ByteBuffer, int[]) se buffer tiver um formato diferente.

Importante: O buffer carregado é uma referência. NÃO MODIFIQUE. Não criamos uma cópia aqui por questões de desempenho, mas se alguma modificação for necessária, faça uma cópia.

Para obter o melhor desempenho, sempre carregue um ByteBuffer direto ou um ByteBuffer apoiado por um array.

Se o buffer for somente leitura, adotamos uma estratégia de cópia na gravação para desempenho.

Carrega um buffer de bytes neste TensorBuffer com formato específico.

Importante: O buffer carregado é uma referência. NÃO MODIFIQUE. Não criamos uma cópia aqui por questões de desempenho, mas se alguma modificação for necessária, faça uma cópia.

Para obter o melhor desempenho, sempre carregue um ByteBuffer direto ou um ByteBuffer apoiado por um array.