tensorflow:: ops:: SparseSoftmax

#include <sparse_ops.h>

Stosuje softmax do wsadowego obiektu ND SparseTensor .

Streszczenie

Dane wejściowe reprezentują ND SparseTensor o logicznym kształcie [..., B, C] (gdzie N >= 2 ) i indeksach posortowanych w kanonicznej kolejności leksykograficznej.

Ta operacja jest równoważna zastosowaniu normalnego tf.nn.softmax() do każdej najgłębszej podmacierzy logicznej o kształcie [B, C] , ale z tym haczykiem, że domyślnie zerowe elementy nie biorą w nich udziału . Dokładniej, algorytm jest równoważny z poniższym:

(1) Stosuje tf.nn.softmax() do zagęszczonego widoku każdej najbardziej wewnętrznej podmacierzy o kształcie [B, C] wzdłuż wymiaru rozmiaru C; (2) Maskuje oryginalne lokalizacje niejawnie zerowe; (3) Renormalizuje pozostałe elementy.

Dlatego wynik SparseTensor ma dokładnie takie same niezerowe indeksy i kształt.

Argumenty:

  • zakres: obiekt zakresu
  • sp_indices: 2-wymiarowa macierz NNZ x R z indeksami wartości niepustych w SparseTensor, w kolejności kanonicznej.
  • sp_values: 1-D. NNZ niepuste wartości odpowiadające sp_indices .
  • sp_shape: 1-D. Kształt wejściowego obiektu SparseTensor.

Zwroty:

  • Output : 1-D. Wartości NNZ dla wyniku SparseTensor .

Konstruktory i destruktory

SparseSoftmax (const :: tensorflow::Scope & scope, :: tensorflow::Input sp_indices, :: tensorflow::Input sp_values, :: tensorflow::Input sp_shape)

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

SparseSoftmax 

 SparseSoftmax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input sp_indices,
  ::tensorflow::Input sp_values,
  ::tensorflow::Input sp_shape
)

węzeł 

::tensorflow::Node * node() const

operator::tensorflow::Input 

 operator::tensorflow::Input() const

operator::tensorflow::Output 

 operator::tensorflow::Output() const