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.

Méthodes publiques

Sortie <T>
comme Sortie ()
Renvoie le handle symbolique d'un tenseur.
statique <T, U étend le nombre> ExpandDims <T>
créer ( portée Scope , entrée opérande <T>, axe opérande <U>)
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

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)]`.
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.