@frozen
public struct LayerNorm<Scalar> : Layer where Scalar : TensorFlowFloatingPoint
Une couche qui applique la normalisation des couches sur un mini-lot d'entrées.
Référence : Normalisation des couches .
La valeur de décalage, également appelée bêta.
Déclaration
public var offset: Tensor<Scalar>
La valeur d'échelle, également connue sous le nom de gamma.
Déclaration
public var scale: Tensor<Scalar>
L'axe.
Déclaration
@noDerivative public let axis: Int
La valeur epsilon de la variance.
Déclaration
@noDerivative public let epsilon: Scalar
Crée un calque de normalisation de calque.
Déclaration
public init( featureCount: Int, axis: Int, epsilon: Scalar = 0.001 )
Paramètres
featureCount
Le nombre de fonctionnalités.
axis
L'axe qui doit être normalisé.
epsilon
Le petit scalaire ajouté à la variance.