tensorflow:: ops:: FusedPadConv2D
#include <nn_ops.h>Realiza un relleno como preprocesamiento durante una convolución.
Resumen
De forma similar a FusedResizeAndPadConv2d, esta operación permite una implementación optimizada donde la etapa de transformación de relleno espacial se fusiona con la búsqueda im2col, pero en este caso sin el filtrado bilineal necesario para el redimensionamiento. La fusión del relleno evita la necesidad de escribir los resultados intermedios como tensores completos, lo que reduce la presión de memoria, y podemos obtener algunas mejoras en la latencia al fusionar los cálculos de transformación. El atributo data_format para Conv2D no es compatible con esta operación, y en su lugar se utiliza el orden 'NHWC'. Internamente, esta operación utiliza un único búfer temporal por grafo, lo que significa que se bloqueará si se ejecutan varias versiones en paralelo. Esto se debe a que este operador es principalmente una optimización para minimizar el uso de memoria.
Argumentos:
- ámbito: Un objeto de ámbito
- Entrada: 4-D con forma
[batch, in_height, in_width, in_channels]. - rellenos: Una matriz de dos columnas que especifica los tamaños de relleno. El número de filas debe ser el mismo que el rango de
input. - Filtro: 4-D con forma
[filter_height, filter_width, in_channels, out_channels]. - pasos: 1-D de longitud 4. El paso de la ventana deslizante para cada dimensión de
input. Debe estar en el mismo orden que la dimensión especificada con formato. - relleno: El tipo de algoritmo de relleno que se utilizará.
Devoluciones:
-
Output: El tensor de salida.
Constructores y destructores | |
|---|---|
FusedPadConv2D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input paddings, :: tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding) |
Atributos públicos | |
|---|---|
operation | |
output | |
Funciones públicas | |
|---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const | |
Atributos públicos
operación
Operation operation
producción
::tensorflow::Output output
Funciones públicas
FusedPadConv2D
FusedPadConv2D( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input paddings, ::tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding )
nodo
::tensorflow::Node * node() const
operador::tensorflow::Input
operator::tensorflow::Input() const
operador::tensorflow::Salida
operator::tensorflow::Output() const