InstanceNorm

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

Uma camada que aplica a normalização de instância em um minilote de entradas.

Referência: Instância Normalização: o ingrediente que falta para estilização rápido .

  • O valor de deslocamento, também conhecido como beta.

    Declaração

    public var offset: Tensor<Scalar> { get set }
  • O valor da escala, também conhecido como gama.

    Declaração

    public var scale: Tensor<Scalar> { get set }
  • O eixo onde estão os recursos.

    Declaração

    public var axis: Int { get }
  • O valor épsilon da variação.

    Declaração

    public var epsilon: Scalar { get }
  • Cria uma camada de normalização de instância.

    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 deslocamento e a escala devem ter o mesmo formato.

    Declaração

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

    Parâmetros

    offset

    O valor de deslocamento inicial.

    scale

    O valor da escala inicial.

    axis

    O eixo onde estão os recursos.

    epsilon

    O valor épsilon da variação.

  • Cria uma camada de normalização de instância.

    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.

    Declaração

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

    Parâmetros

    featureCount

    O número de recursos.

    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.

    Declaração

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

    Parâmetros

    input

    A entrada para a camada.

    Valor de retorno

    A saída.