Norme de groupe

@frozen
public struct GroupNorm<Scalar> : Layer where Scalar : TensorFlowFloatingPoint

Une couche qui applique la normalisation de groupe sur un mini-lot d'entrées.

Référence : Normalisation de groupe .

• compenser
  La valeur de décalage, également appelée bêta.

  Déclaration

  public var offset: Tensor<Scalar>

• échelle
  La valeur d'échelle, également connue sous le nom de gamma.

  Déclaration

  public var scale: Tensor<Scalar>

• nombre de groupes
  Le nombre de groupes.

  Déclaration

  @noDerivative
  public let groupCount: Int

• axe
  L'axe où se trouvent les entités.

  Déclaration

  @noDerivative
  public let axis: Int

• épsilon
  La valeur epsilon de la variance.

  Déclaration

  @noDerivative
  public let epsilon: Scalar

• init(offset:scale:groupCount:axis:epsilon :)
  Crée une couche de normalisation de groupe.

  Condition préalable

  L'axe ne peut pas être un axe par lots.

  Condition préalable

  Le décalage doit être de rang 1.

  Condition préalable

  Le nombre d'éléments du décalage doit être divisible par groupes.

  Condition préalable

  Le décalage et l'échelle doivent avoir la même forme.

  Déclaration

  public init(
    offset: Tensor<Scalar>,
    scale: Tensor<Scalar>,
    groupCount: Int,
    axis: Int,
    epsilon: Scalar
  )

  Paramètres

  offset	La valeur de décalage initiale.
  scale	La valeur d'échelle initiale.
  groupCount	Le nombre de groupes.
  axis	L'axe où se trouvent les entités.
  epsilon	La valeur epsilon de la variance.

• init (featureCount:groupCount:axis:epsilon :)
  Crée une couche de normalisation de groupe.

  Condition préalable

  L'axe ne peut pas être un axe par lots.

  Condition préalable

  Le nombre de fonctionnalités doit être divisible par groupes.

  Déclaration

  public init(
    featureCount: Int,
    groupCount: Int,
    axis: Int = -1,
    epsilon: Scalar = 1e-3
  )

  Paramètres

  featureCount	Le nombre de fonctionnalités.
  groupCount	Le nombre de groupes.
  axis	L'axe où se trouvent les entités. La valeur par défaut est -1.
  epsilon	Le petit scalaire ajouté à la variance. La valeur par défaut est 0,001.

• avant(_:)
  Renvoie le résultat obtenu en appliquant le calque à l’entrée donnée.

  Condition préalable

  L'axe ne peut pas être un axe par lots.

  Condition préalable

  Le nombre d'entités de l'entrée et le décalage doivent être identiques.

  Déclaration

  @differentiable
  public func forward(_ input: Tensor<Scalar>) -> Tensor<Scalar>

  Paramètres

  input

  L'entrée dans le calque.

  Valeur de retour

  La sortie.