przepływ tensorowy:: ops:: SparseSoftmax

#include <sparse_ops.h>

Stosuje softmax do wsadowego ND SparseTensor .

Streszczenie

Dane wejściowe reprezentują ND SparseTensor o kształcie logicznym [..., B, C] (gdzie N >= 2 ) i indeksach posortowanych w kanonicznym porządku leksykograficznym.

Ta operacja jest równoważna zastosowaniu normalnej funkcji tf.nn.softmax() do każdej najbardziej wewnętrznej podmacierzy logicznej o kształcie [B, C] , ale z haczykiem, że domyślnie zerowe elementy nie biorą udziału . W szczególności algorytm jest równoważny następującemu:

(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 z domyślnym zerem; (3) Renormalizuje pozostałe elementy.

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

Argumenty:

  • zakres: Obiekt Scope
  • sp_indices: 2-D. Macierz NNZ x R z indeksami wartości niepustych w SparseTensorze, w porządku kanonicznym.
  • sp_values: 1-D. NNZ niepuste wartości odpowiadające sp_indices .
  • sp_shape: 1-D. Kształt wejścia 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

operacja

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::Wejście

 operator::tensorflow::Input() const 

operator::tensorflow::Wyjście

 operator::tensorflow::Output() const