@frozen
public struct GroupNorm<Scalar> : Layer where Scalar : TensorFlowFloatingPoint
Uma camada que aplica a normalização de grupo em um minilote de entradas.
Referência: Grupo A normalização .
O valor de deslocamento, também conhecido como beta.
Declaração
public var offset: Tensor<Scalar>
O valor da escala, também conhecido como gama.
Declaração
public var scale: Tensor<Scalar>
O número de grupos.
Declaração
@noDerivative public let groupCount: Int
O eixo onde estão os recursos.
Declaração
@noDerivative public let axis: Int
O valor épsilon da variação.
Declaração
@noDerivative public let epsilon: Scalar
Cria uma camada de normalização de grupo.
Condição prévia
O eixo não pode ser um eixo de lote.Condição prévia
O deslocamento deve ter classificação 1.Condição prévia
O número de elementos do deslocamento deve ser divisível por grupos.Condição prévia
O deslocamento e a escala devem ter o mesmo formato.Declaração
Parâmetros
offset
O valor de deslocamento inicial.
scale
O valor da escala inicial.
groupCount
O número de grupos.
axis
O eixo onde estão os recursos.
epsilon
O valor épsilon da variação.
Cria uma camada de normalização de grupo.
Condição prévia
O eixo não pode ser um eixo de lote.Condição prévia
A contagem de recursos deve ser divisível por grupos.Declaração
public init( featureCount: Int, groupCount: Int, axis: Int = -1, epsilon: Scalar = 1e-3 )
Parâmetros
featureCount
O número de recursos.
groupCount
O número de grupos.
axis
O eixo onde estão os recursos. O valor padrão é -1.
epsilon
O pequeno escalar adicionado à variância. O valor padrão é 0,001.
Retorna a saída obtida da aplicação da camada à entrada fornecida.
Condição prévia
O eixo não pode ser um eixo de lote.Condição prévia
O número de recursos da entrada e o deslocamento devem ser os mesmos.Parâmetros
input
A entrada para a camada.
Valor de retorno
A saída.