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]
`-1-input.dims() <= dim <= input.dims()`
Cette opération est liée à `squeeze()`, qui supprime les dimensions de taille 1.
Méthodes publiques
| Sortie <T> | comme Sortie () Renvoie le handle symbolique d'un tenseur. |
| statique <T, U étend le nombre> 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
Méthodes publiques
sortie publique <T> asOutput ()
Renvoie le handle symbolique d'un 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 , entrée opérande <T>, axe opérande <U>)
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)]`. |
Retours
- 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.