TensorBuffer

classe abstraite publique TensorBuffer
Sous-classes directes connues

Représente le tampon de données pour l'entrée ou la sortie d'un modèle.

Méthodes publiques

statique TensorBuffer
createDynamic(DataType dataType)
Crée une classe TensorBuffer dynamique vide avec la valeur DataType spécifiée.
statique TensorBuffer
createFixedSize(int[] shape, DataType dataType)
Crée un TensorBuffer avec les valeurs shape et DataType spécifiées.
statique TensorBuffer
createFrom(tampon TensorBuffer, type de données DataType)
Crée un objet TensorBuffer en effectuant une copie approfondie des données d'une autre instance, avec la valeur DataType spécifiée.
ByteBuffer
getBuffer()
Renvoie le tampon de données.
abstrait DataType
getDataType()
Renvoie le type de données de ce tampon.
int
getFlatSize()
Récupère la valeur flatSize du tampon.
abstrait float[]
getFloatArray()
Renvoie un tableau à virgule flottante des valeurs stockées dans ce tampon.
abstrait se superposer
getFloatValue(int absIndex)
Renvoie une valeur flottante à un indice donné.
abstrait entier[]
getIntArray()
Renvoie un tableau int des valeurs stockées dans ce tampon.
abstrait entier
getIntValue(int absIndex)
Renvoie une valeur int à un index donné.
entier[]
getShape()
Récupère la forme actuelle.
abstrait entier
getTypeSize()
Renvoie le nombre d'octets d'un seul élément du tableau.
booléen
isDynamic()
Renvoie si l'TensorBuffer est de taille dynamique (elle pourrait être redimensionnée arbitrairement).
abstrait vide
loadArray(int[] src, int[] shape)
Charge un tableau int dans ce tampon avec une forme spécifique.
abstrait vide
loadArray(float[] src, int[] shape)
Charge un tableau de valeurs flottantes avec une forme spécifique dans ce tampon.
vide
loadArray(float[] src)
Charge un tableau à virgule flottante dans ce tampon.
vide
loadArray(int[] src)
Charge un tableau int dans ce tampon.
vide
loadBuffer(tampon ByteBuffer)
Charge un tampon d'octets dans ce TensorBuffer.
vide
loadBuffer(tampon ByteBuffer, forme int[])
Charge un tampon d'octets dans cet élément TensorBuffer avec une forme spécifique.

Méthodes héritées

Méthodes publiques

<ph type="x-smartling-placeholder"></ph> public statique TensorBuffer createDynamic (DataType type de données)

Crée une classe TensorBuffer dynamique vide avec la valeur DataType spécifiée. La forme du le TensorBuffer créé est {0}.

Les TensorBuffers dynamiques réallouez de la mémoire lors du chargement de tableaux ou de tampons de données 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.
 

Paramètres
dataType Le type de données du TensorBuffer à créer.

<ph type="x-smartling-placeholder"></ph> public statique TensorBuffer createFixedSize (int[] shape, DataType dataType)

Crée un TensorBuffer avec les valeurs shape et DataType spécifiées. Voici quelques exemples 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);
 

Une fois créé, un TensorBuffer de taille fixe ne peut plus être modifié.

Paramètres
shape Forme du TensorBuffer à créer.
dataType Le type de données du TensorBuffer à créer.