przepływ tensorowy:: ops:: SparseSoftmaxCrossEntropia z logami
#include <nn_ops.h>
Oblicza koszt entropii krzyżowej softmax i gradienty do propagacji wstecznej.
Streszczenie
W przeciwieństwie do SoftmaxCrossEntropyWithLogits
, ta operacja nie akceptuje macierzy prawdopodobieństw etykiet, ale raczej pojedynczą etykietę na wiersz cech. Uważa się, że ta etykieta ma prawdopodobieństwo 1,0 dla danego wiersza.
Dane wejściowe to logity, a nie prawdopodobieństwa.
Argumenty:
- zakres: Obiekt zakresu
- cechy: wielkość_partii x liczba_klas macierz
- etykiety: wektor batch_size z wartościami w [0, num_classes). To jest etykieta dla danego wpisu minipartii.
Zwroty:
- Strata
Output
: Na przykład strata (wektor Batch_size). - Backprop
Output
: gradienty wstecznie propagowane (matryca 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
SparseSoftmaxCrossEntropia z logami
SparseSoftmaxCrossEntropyWithLogits( const ::tensorflow::Scope & scope, ::tensorflow::Input features, ::tensorflow::Input labels )