TensorBuffer

Crée un TensorBuffer dynamique vide avec DataType spécifié. La forme du TensorBuffer créé est {0}.

Dynamic TensorBuffers réallouera la mémoire lors du chargement de tableaux ou de tampons de données de différentes tailles de tampon. Voici quelques exemples :

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

Crée un TensorBuffer avec shape et DataType spécifiés. Voici quelques exemples :

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

La taille d'un TensorBuffer de taille fixe ne peut pas être modifiée une fois qu'il est créé.

Crée un TensorBuffer copiant en profondeur les données d'un autre, avec DataType spécifié.

Renvoie le tampon de données.

Renvoie le type de données de ce tampon.

Obtient la taille plate du tampon.

Renvoie un tableau flottant des valeurs stockées dans ce tampon. Si le tampon est de types différents de float, les valeurs seront converties en float. Par exemple, les valeurs de TensorBufferUint8 seront converties de uint8 en float.

Renvoie une valeur flottante à un index donné. Si le tampon est de types différents de float, la valeur sera convertie en float. Par exemple, lors de la lecture d'une valeur à partir de TensorBufferUint8 , la valeur sera d'abord lue sous la forme uint8, puis sera convertie de uint8 en 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);
 

Renvoie un tableau int des valeurs stockées dans ce tampon. Si le tampon est d'un type différent de int, les valeurs seront converties en int et une perte de précision peut s'appliquer. Par exemple, en obtenant un tableau int à partir d'un TensorBufferFloat avec les valeurs {400.32f, 23.04f}, le résultat est {400, 23}.

Renvoie une valeur int à un index donné. Si le tampon est de types différents de int, la valeur sera convertie en int. Par exemple, lors de la lecture d'une valeur à partir de TensorBufferFloat , la valeur sera d'abord lue sous forme de float, puis sera convertie de float en int. Une perte de précision peut s'appliquer.

 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.
 

Obtient la forme actuelle. (renvoyer une copie ici pour éviter toute modification inattendue.)

Renvoie le nombre d'octets d'un seul élément du tableau. Par exemple, un tampon flottant renverra 4 et un tampon d'octets renverra 1.

Renvoie si le TensorBuffer est de taille dynamique (peut être redimensionné arbitrairement).

Charge un tableau int dans ce tampon avec une forme spécifique. Si le tampon est de type différent de int, les valeurs seront converties dans le type du tampon avant d'être chargées dans le tampon, et une perte de précision peut s'appliquer. Par exemple, en chargeant un tableau int avec des valeurs {400, -23} dans un TensorBufferUint8 , les valeurs seront limitées à [0, 255] puis converties en uint8 par {255, 0}.

Charge un tableau float dans ce tampon avec une forme spécifique. Si le tampon est d'un type différent de celui de float, les valeurs seront converties dans le type du tampon avant d'être chargées dans le tampon, et une perte de précision peut s'appliquer. Par exemple, en chargeant un tableau float dans un TensorBufferUint8 avec des valeurs {400.32f, -23.04f}, les valeurs seront limitées à [0, 255] puis converties en uint8 par {255, 0}.

Charge un tableau float dans ce tampon. Si le tampon est d'un type différent de celui de float, les valeurs seront converties dans le type du tampon avant d'être chargées dans le tampon, et une perte de précision peut s'appliquer. Par exemple, en chargeant un tableau float dans un TensorBufferUint8 avec des valeurs {400.32f, -23.04f}, les valeurs seront limitées à [0, 255] puis converties en uint8 par {255, 0}.

L'utilisation de cette méthode suppose que la forme de src est la même que la forme de ce TensorBuffer . Ainsi, la taille du buffer ( src.length ) doit toujours correspondre à la taille plate de ce TensorBuffer , pour TensorBuffer de taille fixe et dynamique. Utilisez loadArray(float[], int[]) si src a une forme différente.

Charge un tableau int dans ce tampon. Si le tampon est de type différent de int, les valeurs seront converties dans le type du tampon avant d'être chargées dans le tampon, et une perte de précision peut s'appliquer. Par exemple, en chargeant un tableau int avec des valeurs {400, -23} dans un TensorBufferUint8 , les valeurs seront limitées à [0, 255] puis converties en uint8 par {255, 0}.

L'utilisation de cette méthode suppose que la forme de src est la même que la forme de ce TensorBuffer . Ainsi, la taille du buffer ( src.length ) doit toujours correspondre à la taille plate de ce TensorBuffer , pour TensorBuffer de taille fixe et dynamique. Utilisez loadArray(int[], int[]) si src a une forme différente.

Charge un tampon d'octets dans ce TensorBuffer . La taille du tampon doit correspondre à la taille plate de ce TensorBuffer .

L'utilisation de cette méthode suppose que la forme du buffer est la même que la forme de ce TensorBuffer . Ainsi, la taille du buffer ( buffer.limit() ) doit toujours correspondre à la taille plate de ce TensorBuffer , pour TensorBuffer de taille fixe et dynamique. Utilisez loadBuffer(ByteBuffer, int[]) si buffer a une forme différente.

Important : Le tampon chargé est une référence. NE PAS MODIFIER. Nous ne créons pas de copie ici pour des raisons de performances, mais si une modification est nécessaire, veuillez en faire une copie.

Pour de meilleures performances, chargez toujours un ByteBuffer direct ou un ByteBuffer soutenu par un tableau.

Si le buffer est en lecture seule, nous adoptons une stratégie de copie sur écriture pour les performances.

Charge un tampon d'octets dans ce TensorBuffer avec une forme spécifique.

Important : Le tampon chargé est une référence. NE PAS MODIFIER. Nous ne créons pas de copie ici pour des raisons de performances, mais si une modification est nécessaire, veuillez en faire une copie.

Pour de meilleures performances, chargez toujours un ByteBuffer direct ou un ByteBuffer soutenu par un tableau.