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