flujo tensor:: operaciones:: FusionadoPadConv2D
#include <nn_ops.h>
Realiza un relleno como preproceso durante una convolución.
Resumen
Similar a FusedResizeAndPadConv2d, esta operación permite una implementación optimizada donde la etapa de transformación del relleno espacial se fusiona con la búsqueda im2col, pero en este caso sin el filtrado bilineal necesario para cambiar el tamaño. Fusionar el relleno evita la necesidad de escribir los resultados intermedios como tensores completos, lo que reduce la presión de la memoria, y podemos obtener algunas ganancias de latencia fusionando los cálculos de transformación. Esta operación no admite el atributo data_format para Conv2D y en su lugar se utiliza el orden 'NHWC'. Internamente, esta operación utiliza un único búfer temporal por gráfico, 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:
- alcance: un objeto de alcance
- entrada: 4-D con forma
[batch, in_height, in_width, in_channels]
. - paddings: 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]
. - zancadas: 1-D de longitud 4. La zancada de la ventana deslizante para cada dimensión de
input
. Debe estar en el mismo orden que la dimensión especificada con formato. - padding: 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
FusionadoPadConv2D
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::Entrada
operator::tensorflow::Input() const
operador::tensorflow::Salida
operator::tensorflow::Output() const