tensorflow:: ops:: FusedPadConv2D

#include <nn_ops.h>

Wykonuje wypełnianie jako proces wstępny podczas splotu.

Streszczenie

Podobnie jak FusedResizeAndPadConv2d, ta operacja umożliwia zoptymalizowaną implementację, w której etap transformacji wypełnienia przestrzennego jest łączony z wyszukiwaniem im2col, ale w tym przypadku bez filtrowania dwuliniowego wymaganego do zmiany rozmiaru. Połączenie wypełnienia eliminuje konieczność zapisywania wyników pośrednich jako całych tensorów, zmniejszając obciążenie pamięci, a scalając obliczenia transformacji, możemy uzyskać pewne korzyści w zakresie opóźnień. Atrybut data_format dla Conv2D nie jest obsługiwany przez tę operację i zamiast niego używana jest kolejność „NHWC”. Wewnętrznie ta operacja używa pojedynczego bufora roboczego dla każdego wykresu, co oznacza, że ​​zablokuje się, jeśli wiele wersji będzie uruchamianych równolegle. Dzieje się tak, ponieważ ten operator jest przede wszystkim optymalizacją mającą na celu minimalizację zużycia pamięci.

Argumenty:

  • zakres: obiekt zakresu
  • wejście: 4-D z kształtem [batch, in_height, in_width, in_channels] .
  • Wypełnienia: Dwukolumnowa macierz określająca rozmiary wypełnień. Liczba wierszy musi być taka sama jak ranga input .
  • filtr: 4-D o kształcie [filter_height, filter_width, in_channels, out_channels] .
  • Kroki: 1-D o długości 4. Krok okna przesuwnego dla każdego wymiaru danych input . Musi być w tej samej kolejności, co wymiar określony w formacie.
  • padding: Typ algorytmu wypełniania, który ma zostać użyty.

Zwroty:

Konstruktory i destruktory

FusedPadConv2D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input paddings, :: tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding)

Atrybuty publiczne

operation
output

Funkcje publiczne

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Atrybuty publiczne

działanie

Operation operation

wyjście

::tensorflow::Output output

Funkcje publiczne

FusedPadConv2D

 FusedPadConv2D(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input paddings,
  ::tensorflow::Input filter,
  StringPiece mode,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding
)

węzeł

::tensorflow::Node * node() const 

operator::tensorflow::Input

 operator::tensorflow::Input() const 

operator::tensorflow::Output

 operator::tensorflow::Output() const