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 Rcom os índices dos valores não vazios em um SparseTensor, em ordenação canônica. - sp_values: 1-D.
NNZvalores não vazios correspondentes asp_indices. - sp_shape: 1-D. Formato do SparseTensor de entrada.
Devoluções:
-
Output: 1-D. Os valoresNNZpara oSparseTensorresultante.
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