przepływ tensorowy:: ops:: SoftmaxCrossEntropyWithLogits

#include <nn_ops.h>

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

Streszczenie

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

Argumenty:

  • zakres: Obiekt Scope
  • cechy: macierz rozmiar_wsadu x liczba_klas
  • etykiety: rozmiar_wsadu x macierz num_classes Osoba wywołująca musi upewnić się, że każda partia etykiet reprezentuje prawidłowy rozkład prawdopodobieństwa.

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

SoftmaxCrossEntropyWithLogits (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

SoftmaxCrossEntropyWithLogits

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