@frozen
public struct GroupNorm<Scalar> : Layer where Scalar : TensorFlowFloatingPoint
Una capa que aplica la normalización de grupo sobre un mini lote de entradas.
Referencia: Normalización de grupos .
El valor de compensación, también conocido como beta.
Declaración
public var offset: Tensor<Scalar>
El valor de escala, también conocido como gamma.
Declaración
public var scale: Tensor<Scalar>
El número de grupos.
Declaración
@noDerivative public let groupCount: Int
El eje donde se encuentran las características.
Declaración
@noDerivative public let axis: Int
El valor épsilon de varianza.
Declaración
@noDerivative public let epsilon: Scalar
Crea una capa de normalización de grupo.
Condición previa
El eje no puede ser un eje por lotes.Condición previa
La compensación debe tener rango 1.Condición previa
El número de elementos del desplazamiento debe ser divisible por grupos.Condición previa
El desplazamiento y la escala deben tener la misma forma.Declaración
Parámetros
offset
El valor de compensación inicial.
scale
El valor de escala inicial.
groupCount
El número de grupos.
axis
El eje donde se encuentran las características.
epsilon
El valor épsilon de varianza.
Crea una capa de normalización de grupo.
Condición previa
El eje no puede ser un eje por lotes.Condición previa
El recuento de funciones debe ser divisible por grupos.Declaración
public init( featureCount: Int, groupCount: Int, axis: Int = -1, epsilon: Scalar = 1e-3 )
Parámetros
featureCount
El número de características.
groupCount
El número de grupos.
axis
El eje donde se encuentran las características. El valor predeterminado es -1.
epsilon
El pequeño escalar aumentó la varianza. El valor predeterminado es 0,001.
Devuelve el resultado obtenido al aplicar la capa a la entrada dada.
Condición previa
El eje no puede ser un eje por lotes.Condición previa
El número de características de la entrada y el desplazamiento deben ser los mismos.Parámetros
input
La entrada a la capa.
Valor de retorno
La salida.