TensorBuffer

Tworzy pusty dynamiczny TensorBuffer z określonym DataType . Kształt utworzonego TensorBuffer to {0}.

Dynamiczne TensorBuffers ponownie przydzielą pamięć podczas ładowania tablic lub buforów danych o różnych rozmiarach buforów. Oto kilka przykładów:

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

Tworzy TensorBuffer o określonym shape i DataType . Oto kilka przykładów:

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

Nie można zmienić rozmiaru TensorBuffer o stałym rozmiarze po jego utworzeniu.

Tworzy TensorBuffer głębokie kopiowanie danych z innego, z określonym DataType .

Zwraca bufor danych.

Zwraca typ danych tego bufora.

Pobiera flatSize buforu.

Zwraca tablicę zmiennoprzecinkową wartości przechowywanych w tym buforze. Jeżeli bufor jest innego typu niż float, wartości zostaną przekonwertowane na float. Na przykład wartości w TensorBufferUint8 zostaną przekonwertowane z uint8 na float.

Zwraca wartość zmiennoprzecinkową przy danym indeksie. Jeżeli bufor jest innego typu niż float, wartość zostanie przekonwertowana na float. Na przykład podczas odczytywania wartości z TensorBufferUint8 wartość zostanie najpierw odczytana jako uint8, a następnie zostanie przekonwertowana z uint8 na 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);
 

Zwraca tablicę typu int wartości przechowywanych w tym buforze. Jeśli bufor jest innego typu niż int, wartości zostaną przekonwertowane na int i może wystąpić utrata precyzji. Na przykład pobierając tablicę int z TensorBufferFloat o wartościach {400.32f, 23.04f}, wynikiem jest {400, 23}.

Zwraca wartość int przy danym indeksie. Jeżeli bufor jest innego typu niż int, wartość zostanie przekonwertowana na int. Na przykład podczas odczytywania wartości z TensorBufferFloat wartość zostanie najpierw odczytana jako float, a następnie zostanie przekonwertowana z float na int. Może wystąpić utrata precyzji.

 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.
 

Pobiera bieżący kształt. (zwracając kopię tutaj, aby uniknąć nieoczekiwanych modyfikacji.)

Zwraca liczbę bajtów pojedynczego elementu w tablicy. Na przykład bufor zmiennoprzecinkowy zwróci 4, a bufor bajtowy zwróci 1.

Zwraca, jeśli TensorBuffer ma dynamiczny rozmiar (można dowolnie zmieniać rozmiar).

Ładuje tablicę int do tego bufora o określonym kształcie. Jeśli bufor jest innego typu niż int, wartości zostaną przekonwertowane na typ bufora przed załadowaniem do bufora, co może skutkować utratą precyzji. Na przykład ładując tablicę int z wartościami {400, -23} do TensorBufferUint8 , wartości zostaną ograniczone do [0, 255], a następnie rzutowane na uint8 przez {255, 0}.

Ładuje do tego bufora tablicę zmiennoprzecinkową o określonym kształcie. Jeśli bufor jest innego typu niż float, wartości zostaną przekonwertowane na typ bufora przed załadowaniem do bufora i może wystąpić utrata precyzji. Na przykład ładując tablicę zmiennoprzecinkową do TensorBufferUint8 z wartościami {400.32f, -23.04f}, wartości zostaną ograniczone do [0, 255], a następnie rzutowane na uint8 przez {255, 0}.

Ładuje tablicę float do tego bufora. Jeśli bufor jest innego typu niż float, wartości zostaną przekonwertowane na typ bufora przed załadowaniem do bufora i może wystąpić utrata precyzji. Na przykład ładując tablicę zmiennoprzecinkową do TensorBufferUint8 z wartościami {400.32f, -23.04f}, wartości zostaną ograniczone do [0, 255], a następnie rzutowane na uint8 przez {255, 0}.

Użycie tej metody zakłada, że ​​kształt src jest taki sam jak kształt tego TensorBuffer . Zatem rozmiar buffer ( src.length ) powinien zawsze odpowiadać płaskiemu rozmiarowi tego TensorBuffer , zarówno dla TensorBuffer o stałym rozmiarze, jak i dynamicznym. Użyj metody loadArray(float[], int[]) jeśli src ma inny kształt.

Ładuje tablicę int do tego bufora. Jeśli bufor jest innego typu niż int, wartości zostaną przekonwertowane na typ bufora przed załadowaniem do bufora, co może skutkować utratą precyzji. Na przykład ładując tablicę int z wartościami {400, -23} do TensorBufferUint8 , wartości zostaną ograniczone do [0, 255], a następnie rzutowane na uint8 przez {255, 0}.

Użycie tej metody zakłada, że ​​kształt src jest taki sam jak kształt tego TensorBuffer . Zatem rozmiar buffer ( src.length ) powinien zawsze odpowiadać płaskiemu rozmiarowi tego TensorBuffer , zarówno dla TensorBuffer o stałym rozmiarze, jak i dynamicznym. Użyj metody loadArray(int[], int[]) jeśli src ma inny kształt.

Ładuje bufor bajtowy do tego TensorBuffer . Rozmiar bufora musi odpowiadać płaskiemu rozmiarowi tego TensorBuffer .

Użycie tej metody zakłada, że ​​kształt buffer jest taki sam jak kształt tego TensorBuffer . Zatem rozmiar buffer ( buffer.limit() ) powinien zawsze odpowiadać płaskiemu rozmiarowi tego TensorBuffer , zarówno dla TensorBuffer o stałym rozmiarze, jak i dynamicznego. Użyj metody loadBuffer(ByteBuffer, int[]) jeśli buffer ma inny kształt.

Ważne: załadowany bufor jest odniesieniem. NIE MODYFIKUJ. Nie tworzymy tutaj kopii ze względu na wydajność, ale jeśli konieczna jest modyfikacja, wykonaj kopię.

Aby uzyskać najlepszą wydajność, zawsze ładuj bezpośrednio ByteBuffer lub ByteBuffer wspierany przez tablicę.

Jeśli buffer jest tylko do odczytu, w celu zapewnienia wydajności przyjmujemy strategię kopiowania przy zapisie.

Ładuje bufor bajtowy do tego TensorBuffer o określonym kształcie.

Ważne: załadowany bufor jest odniesieniem. NIE MODYFIKUJ. Nie tworzymy tutaj kopii ze względu na wydajność, ale jeśli konieczna jest modyfikacja, wykonaj kopię.

Aby uzyskać najlepszą wydajność, zawsze ładuj bezpośrednio ByteBuffer lub ByteBuffer wspierany przez tablicę.