przepływ tensorowy:: ops:: SparseSoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

Oblicza koszt i gradienty entropii krzyżowej softmax w celu propagacji wstecznej.

Streszczenie

W przeciwieństwie do SoftmaxCrossEntropyWithLogits ta operacja nie akceptuje macierzy prawdopodobieństw etykiet, ale raczej pojedynczą etykietę na wiersz funkcji. Uważa się, że etykieta ta ma prawdopodobieństwo 1,0 dla danego wiersza.

Dane wejściowe to logity, a nie prawdopodobieństwa.

Argumenty:

  • zakres: Obiekt Scope
  • cechy: macierz rozmiar_wsadu x liczba_klas
  • etykiety: wektor rozmiar_wsadu z wartościami w [0, liczba_klas). To jest etykieta dla danej pozycji minipartii.

Zwroty:

  • Strata Output : Na przykład strata (wektor Batch_size).
  • Output podpórka: gradienty propagowane wstecz (macierz Batch_size x num_classes).

Konstruktory i destruktory

SparseSoftmaxCrossEntropyWithLogits (const :: tensorflow::Scope & scope, :: tensorflow::Input features, :: tensorflow::Input labels)

Atrybuty publiczne

backprop
loss
operation

Atrybuty publiczne

podpora

::tensorflow::Output backprop

strata

::tensorflow::Output loss

operacja

Operation operation

Funkcje publiczne

SparseSoftmaxCrossEntropyWithLogits

 SparseSoftmaxCrossEntropyWithLogits(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input features,
  ::tensorflow::Input labels
)