fluxo tensor:: ops:: FusedPadConv2D
#include <nn_ops.h>
Executa um preenchimento como pré-processo durante uma convolução.
Resumo
Semelhante a FusedResizeAndPadConv2d, esta operação permite uma implementação otimizada onde o estágio de transformação de preenchimento espacial é fundido com a pesquisa im2col, mas neste caso sem a filtragem bilinear necessária para redimensionar. A fusão do preenchimento evita a necessidade de escrever os resultados intermediários como tensores inteiros, reduzindo a pressão da memória, e podemos obter alguns ganhos de latência mesclando os cálculos de transformação. O atributo data_format para Conv2D não é suportado por esta operação e a ordem 'NHWC' é usada em seu lugar. Internamente, esta operação usa um único buffer de rascunho por gráfico, o que significa que será bloqueado se várias versões estiverem sendo executadas em paralelo. Isso ocorre porque esse operador é principalmente uma otimização para minimizar o uso de memória.
Argumentos:
- escopo: um objeto Escopo
- entrada: 4-D com forma
[batch, in_height, in_width, in_channels]
. - preenchimentos: uma matriz de duas colunas especificando os tamanhos de preenchimento. O número de linhas deve ser igual à classificação de
input
. - filtro: 4-D com forma
[filter_height, filter_width, in_channels, out_channels]
. - passadas: 1-D de comprimento 4. A passada da janela deslizante para cada dimensão de
input
. Deve estar na mesma ordem que a dimensão especificada com formato. - preenchimento: O tipo de algoritmo de preenchimento a ser usado.
Retorna:
-
Output
: o tensor de saída.
Construtores e Destruidores | |
---|---|
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 |
Funções públicas | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Atributos públicos
operação
Operation operation
saída
::tensorflow::Output output
Funções 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 )
nó
::tensorflow::Node * node() const
operador::tensorflow::Input
operator::tensorflow::Input() const
operador::tensorflow::Saída
operator::tensorflow::Output() const