Shape

classe finale publique

La forme peut-être partiellement connue d'un tenseur produit par une opération.

Méthodes publiques

booléen
est égal (Obj objet)
int
Forme statique
make (long firstDimensionSize, long... otherDimensionSizes)
Créez une forme représentant une valeur à N dimensions.
int
numDimensions ()
Nombre de dimensions représentées par cette forme.
Forme statique
scalaire ()
Créez une forme représentant une valeur scalaire.
long
taille (int i)
La taille de la i-ème dimension.
Chaîne
àChaîne ()
Description succincte de la forme destinée au débogage.
Forme statique
inconnu ()
Créez une forme représentant un nombre inconnu de dimensions.

Méthodes héritées

Méthodes publiques

public booléen égal (Objet obj)

Paramètres
obj

code de hachage int public ()

marque de forme statique publique (long firstDimensionSize, long... otherDimensionSizes)

Créez une forme représentant une valeur à N dimensions.

Crée une forme représentant une valeur à N dimensions (N étant au moins 1), avec la taille fournie pour chaque dimension. A -1 indique que la taille de la dimension correspondante est inconnue. Par exemple :

// A 2-element vector.
 Shape vector = Shape.create(2);

 // A 2x3 matrix.
 Shape matrix = Shape.create(2, 3);

 // A matrix with 4 columns but an unknown number of rows.
 // This is typically used to indicate the shape of tensors that represent
 // a variable-sized batch of values. The Shape below might represent a
 // variable-sized batch of 4-element vectors.
 Shape batch = Shape.create(-1, 4);
 

Paramètres
premièreDimensionTaille
autresDimensionTailles

public int numDimensions ()

Nombre de dimensions représentées par cette forme.

Retour
  • -1 si le nombre de dimensions est inconnu, 0 si la forme représente un scalaire, 1 pour un vecteur, 2 pour une matrice etc.

Scalaire de forme statique public ()

Créez une forme représentant une valeur scalaire.

taille longue publique (int i)

La taille de la i-ième dimension.

Paramètres
je
Retour
  • La taille de la dimension demandée ou -1 si elle est inconnue.

chaîne publique versString ()

Description succincte de la forme destinée au débogage.

forme statique publique inconnue ()

Créez une forme représentant un nombre inconnu de dimensions.