przepływ tensorowy:: ops:: FusedPadConv2D
#include <nn_ops.h>
Wykonuje dopełnienie jako proces wstępny podczas splotu.
Streszczenie
Podobnie jak w przypadku FusedResizeAndPadConv2d, ta opcja pozwala na zoptymalizowaną implementację, w której etap transformacji przestrzennego dopełnienia jest połączony z wyszukiwaniem im2col, ale w tym przypadku bez filtrowania dwuliniowego wymaganego do zmiany rozmiaru. Połączenie dopełnienia zapobiega konieczności zapisywania wyników pośrednich jako całych tensorów, co zmniejsza obciążenie pamięci, a dzięki połączeniu obliczeń transformacji możemy uzyskać pewne zyski w zakresie opóźnień. Atrybut data_format dla Conv2D nie jest obsługiwany przez tę operację i zamiast tego używana jest kolejność „NHWC”. Wewnętrznie ta operacja wykorzystuje pojedynczy bufor magazynujący dla każdego wykresu, co oznacza, że będzie blokować, jeśli wiele wersji będzie uruchamianych równolegle. Dzieje się tak, ponieważ ten operator jest przede wszystkim optymalizacją mającą na celu zminimalizowanie zużycia pamięci.
Argumenty:
- zakres: Obiekt Scope
- wejście: 4-D z kształtem
[batch, in_height, in_width, in_channels]
. - dopełnienia: dwukolumnowa macierz określająca rozmiary dopełnienia. Liczba wierszy musi być taka sama jak ranga danych
input
. - filtr: 4-D z kształtem
[filter_height, filter_width, in_channels, out_channels]
. - kroki: 1-D o długości 4. Krok przesuwanego okna dla każdego wymiaru
input
. Musi być w tej samej kolejności, co wymiar określony w formacie. - dopełnienie: typ algorytmu dopeł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::Wejście
operator::tensorflow::Input() const
operator::tensorflow::Wyjście
operator::tensorflow::Output() const