ExpandDims

classe finale publique ExpandDims

Insère une dimension de 1 dans la forme d'un tenseur.

Étant donné un tenseur « entrée », cette opération insère une dimension de 1 à l'index de dimension « axe » de la forme de « entrée ». L'index de dimension « axe » commence à zéro ; si vous spécifiez un nombre négatif pour « axe », il est compté à rebours à partir de la fin.

Cette opération est utile si vous souhaitez ajouter une dimension de lot à un seul élément. Par exemple, si vous avez une seule image de forme `[hauteur, largeur, canaux]`, vous pouvez en faire un lot de 1 image avec `expand_dims(image, 0)`, ce qui rendra la forme `[1, hauteur , largeur, canaux]`.

Autres exemples :

# 't' is a tensor of shape [2]
 shape(expand_dims(t, 0)) ==> [1, 2]
 shape(expand_dims(t, 1)) ==> [2, 1]
 shape(expand_dims(t, -1)) ==> [2, 1]
 
 # 't2' is a tensor of shape [2, 3, 5]
 shape(expand_dims(t2, 0)) ==> [1, 2, 3, 5]
 shape(expand_dims(t2, 2)) ==> [2, 3, 1, 5]
 shape(expand_dims(t2, 3)) ==> [2, 3, 5, 1]
 
Cette opération nécessite que :

`-1-input.dims() <= dim <= input.dims()`

Cette opération est liée à `squeeze()`, qui supprime les dimensions de taille 1.

Constantes

Chaîne OP_NAME Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Méthodes publiques

Sortie <T>
comme Sortie ()
Renvoie le handle symbolique du tenseur.
statique <T étend TType > ExpandDims <T>
créer (portée de portée , entrée opérande <T>, opérande <? étend l'axe TNumber >)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération ExpandDims.
Sortie <T>
sortir ()
Contient les mêmes données que « input », mais sa forme a une dimension supplémentaire de taille 1 ajoutée.

Méthodes héritées

Constantes

chaîne finale statique publique OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Valeur constante : « ExpandDims »

Méthodes publiques

sortie publique <T> asOutput ()

Renvoie le handle symbolique du tenseur.

Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.

public static ExpandDims <T> créer (portée de portée , entrée opérande <T>, opérande <? étend TNumber > axe)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération ExpandDims.

Paramètres
portée portée actuelle
axe 0-D (scalaire). Spécifie l'index de dimension auquel développer la forme de « entrée ». Doit être compris dans la plage `[-rank(input) - 1, Rank(input)]`.
Retour
  • une nouvelle instance de ExpandDims

sortie publique <T> sortie ()

Contient les mêmes données que « input », mais sa forme a une dimension supplémentaire de taille 1 ajoutée.