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:
-
Output: Tensor wyjściowy.
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