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 asp_indices
. - sp_forma: 1-D. Forma da entrada SparseTensor.
Retorna:
-
Output
: 1-D. Os valoresNNZ
para o resultadoSparseTensor
.
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