fluxo tensor:: ops:: EsparsoSoftmax

#include <sparse_ops.h>

Aplica softmax a um ND SparseTensor em lote.

Resumo

As entradas representam um ND SparseTensor com formato lógico [..., B, C] (onde N >= 2 ) e com índices classificados na ordem lexicográfica canônica.

Esta operação é equivalente a aplicar o tf.nn.softmax() normal a cada submatriz lógica mais interna com forma [B, C] , mas com a desvantagem de que os elementos implicitamente zero não participam . Especificamente, o algoritmo é equivalente ao seguinte:

(1) Aplica tf.nn.softmax() a uma visão densificada de cada submatriz mais interna com forma [B, C] , ao longo da dimensão tamanho C; (2) Oculta as localizações originais implicitamente zero; (3) Renormaliza os elementos restantes.

Conseqüentemente, o resultado do SparseTensor tem exatamente os mesmos índices e formato diferentes de zero.

Argumentos:

  • escopo: um objeto Escopo
  • sp_indices: 2-D. Matriz NNZ x R com os índices de valores não vazios em um SparseTensor, em ordenação canônica.
  • sp_valores: 1-D. Valores não vazios NNZ correspondentes a sp_indices .
  • sp_forma: 1-D. Forma da entrada SparseTensor.

Retorna:

  • Output : 1-D. Os valores NNZ para o resultado SparseTensor .

Construtores e Destruidores

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

Atributos públicos

operation
output

Funções públicas

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Atributos públicos

operação 

Operation operation

saída 

::tensorflow::Output output

Funções públicas

EsparsoSoftmax 

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

nó 

::tensorflow::Node * node() const

operador::tensorflow::Input 

 operator::tensorflow::Input() const

operador::tensorflow::Saída 

 operator::tensorflow::Output() const