przepływ tensorowy:: ops:: MaxPoolWithArgmax

#include <nn_ops.h>

Wykonuje maksymalne łączenie na wejściu i wyprowadza zarówno wartości maksymalne, jak i indeksy.

Streszczenie

Indeksy w argmax są spłaszczane, tak że maksymalna wartość w pozycji [b, y, x, c] staje się spłaszczonym indeksem: (y * width + x) * channels + c jeśli include_batch_in_index ma wartość False; ((b * height + y) * width + x) * channels + c jeśli include_batch_in_index ma wartość True.

Zwracane indeksy są zawsze wyrażone w [0, height) x [0, width) przed spłaszczeniem, nawet jeśli występuje dopełnienie i matematycznie poprawna odpowiedź znajduje się na zewnątrz (albo ujemna, albo zbyt duża). Jest to błąd, ale jego naprawa jest trudna do wykonania w bezpieczny sposób, kompatybilny wstecz, zwłaszcza ze względu na spłaszczenie.

Argumenty:

  • zakres: Obiekt Scope
  • wejście: 4-D z kształtem [batch, height, width, channels] . Wejście do puli.
  • ksize: Rozmiar okna dla każdego wymiaru tensora wejściowego.
  • kroki: krok przesuwanego okna dla każdego wymiaru tensora wejściowego.
  • dopełnienie: typ algorytmu dopełniania, który ma zostać użyty.

Opcjonalne atrybuty (patrz Attrs ):

  • include_batch_in_index: Czy uwzględnić wymiar wsadowy w spłaszczonym indeksie argmax .

Zwroty:

  • Wyjście Output : maksymalny łączony tensor wyjściowy.
  • Output argmax: 4-D. Spłaszczone wskaźniki wartości maksymalnych wybranych dla każdego wyjścia.

Konstruktory i destruktory

MaxPoolWithArgmax (const :: tensorflow::Scope & scope, :: tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding)
MaxPoolWithArgmax (const :: tensorflow::Scope & scope, :: tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding, const MaxPoolWithArgmax::Attrs & attrs)

Atrybuty publiczne

argmax
operation
output

Publiczne funkcje statyczne

IncludeBatchInIndex (bool x)
Targmax (DataType x)

Struktury

tensorflow:: ops:: MaxPoolWithArgmax:: Atrybuty

Opcjonalne moduły ustawiające atrybuty dla MaxPoolWithArgmax .

Atrybuty publiczne

argmax

::tensorflow::Output argmax

operacja

Operation operation

wyjście

::tensorflow::Output output

Funkcje publiczne

MaxPoolWithArgmax

 MaxPoolWithArgmax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  const gtl::ArraySlice< int > & ksize,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding
)

MaxPoolWithArgmax

 MaxPoolWithArgmax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  const gtl::ArraySlice< int > & ksize,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding,
  const MaxPoolWithArgmax::Attrs & attrs
)

Publiczne funkcje statyczne

UwzględnijBatchInIndex

Attrs IncludeBatchInIndex(
  bool x
)

Targmax

Attrs Targmax(
  DataType x
)