tensorflow::ops::SparseSoftmax

#include <sparse_ops.h>

Aplica a função softmax a um SparseTensor ND em lote.

Resumo

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

Esta operação é equivalente a aplicar a função tf.nn.softmax() normal a cada submatriz lógica mais interna com formato [B, C] , mas com a ressalva 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 de tamanho C; (2) Mascara as localizações originais implicitamente zero; (3) Renormaliza os elementos restantes.

Portanto, o resultado SparseTensor tem exatamente os mesmos índices não nulos e a mesma forma.

Argumentos:

  • escopo: Um objeto de escopo
  • sp_indices: Matriz 2D. NNZ x R com os índices dos valores não vazios em um SparseTensor, em ordenação canônica.
  • sp_values: 1-D. NNZ valores não vazios correspondentes a sp_indices .
  • sp_shape: 1-D. Formato do SparseTensor de entrada.

Devoluções:

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

Construtores e Destrutores

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

SparseSoftmax

 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