Cette page a été traduite par l'API Cloud Translation.
Switch to English

tf.nn.moments

Tensorflow 1 Version Voir la source sur GitHub

Calcule la moyenne et la variance de x .

La moyenne et la variance sont calculées en additionnant le contenu de x sur les axes . Si x est 1-D et axes = [0] ceci est juste la moyenne et la variance d'un vecteur.

Lors de l' utilisation de ces moments pour la normalisation des lots (voir tf.nn.batch_normalization ):

  • pour ce qu'on appelle la « normalisation globale », utilisé avec des filtres de convolution avec une forme [batch, height, width, depth] , passer axes=[0, 1, 2] .
  • pour de simples passe de normalisation par lots axes=[0] (lot seulement).

x Un Tensor .
axes Tableau de ints. Le long des axes qui pour calculer moyenne et la variance.
shift Non utilisé dans la mise en œuvre actuelle.
keepdims produire des moments avec le même dimensionnalité que l'entrée.
name Le nom utilisé pour la portée des opérations qui calculent les moments.

Deux Tensor objets: mean et la variance .