Normalización por lotes.
Tenga en cuenta que el tamaño de los tensores 4D está definido por "NHWC" o "NCHW". El tamaño de los tensores 1D coincide con la dimensión C de los tensores 4D.
Clases anidadas
clase | Opciones de FusedBatchNorm | Atributos opcionales para FusedBatchNorm |
Constantes
Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
Salida <U> | loteMedio () Un tensor 1D para la media por lotes calculada, que TensorFlow utilizará para calcular la media en ejecución. |
Salida <U> | variación por lotes () Un tensor 1D para la varianza del lote calculada, que TensorFlow utilizará para calcular la varianza en ejecución. |
estático <T extiende TNumber , U extiende TNumber > FusedBatchNorm <T, U> | crear (alcance alcance , operando <T> x, escala del operando <U>, desplazamiento del operando <U>, media del operando <U>, varianza del operando <U>, opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación FusedBatchNorm. |
Opciones estáticas de FusedBatchNorm | formato de datos (formato de datos de cadena) |
Opciones estáticas de FusedBatchNorm | épsilon (épsilon flotante) |
Opciones estáticas de FusedBatchNorm | exponentialAvgFactor (Flotador exponencialAvgFactor) |
Opciones estáticas de FusedBatchNorm | isTraining (booleano isTraining) |
Salida <U> | reservarEspacio1 () Un tensor 1D para la media del lote calculada, que se reutilizará en el cálculo del gradiente. |
Salida <U> | reservarEspacio2 () Un tensor 1D para la varianza del lote calculada (varianza invertida en el caso de cuDNN), que se reutilizará en el cálculo del gradiente. |
Salida <U> | reservarEspacio3 () Un tensor 1D para algunos resultados intermedios, que se reutilizará en el cálculo del gradiente para una mejor eficiencia. |
Salida <T> | y () Un tensor 4D para datos de salida. |
Métodos heredados
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo conoce el motor central de TensorFlow.
Métodos públicos
Salida pública <U> loteMean ()
Un tensor 1D para la media por lotes calculada, que TensorFlow utilizará para calcular la media en ejecución.
Salida pública <U> variación por lotes ()
Un tensor 1D para la varianza del lote calculada, que TensorFlow utilizará para calcular la varianza en ejecución.
pública estática FusedBatchNorm <T, U> crear (alcance del alcance , operando <T> x, escala del operando <U>, desplazamiento del operando <U>, media del operando <U>, varianza del operando <U>, opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación FusedBatchNorm.
Parámetros
alcance | alcance actual |
---|---|
X | Un tensor 4D para datos de entrada. |
escala | Un tensor 1D para factor de escala, para escalar la x normalizada. |
compensar | Un tensor 1D para desplazamiento, para cambiar a la x normalizada. |
significar | Un tensor 1D para la media poblacional. Se utiliza únicamente para inferencias; debe estar vacío para el entrenamiento. |
diferencia | Un tensor 1D para la varianza de la población. Se utiliza únicamente para inferencias; debe estar vacío para el entrenamiento. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de FusedBatchNorm
FusedBatchNorm.Options estático público formato de datos (formato de datos de cadena)
Parámetros
formato de datos | El formato de datos para x e y. Ya sea "NHWC" (predeterminado) o "NCHW". |
---|
pública estática FusedBatchNorm.Options épsilon (Float épsilon)
Parámetros
épsilon | Un pequeño número flotante agregado a la varianza de x. |
---|
público estático FusedBatchNorm.Options isTraining (booleano isTraining)
Parámetros
está entrenando | Un valor bool para indicar que la operación es para entrenamiento (predeterminado) o inferencia. |
---|
Salida pública <U> reserveSpace1 ()
Un tensor 1D para la media del lote calculada, que se reutilizará en el cálculo del gradiente.
Salida pública <U> reserveSpace2 ()
Un tensor 1D para la varianza del lote calculada (varianza invertida en el caso de cuDNN), que se reutilizará en el cálculo del gradiente.
Salida pública <U> reserveSpace3 ()
Un tensor 1D para algunos resultados intermedios, que se reutilizará en el cálculo del gradiente para una mejor eficiencia.