flujo tensor:: operaciones:: EscasoSoftmax

#include <sparse_ops.h>

Aplica softmax a un ND SparseTensor por lotes.

Resumen

Las entradas representan un ND SparseTensor con forma lógica [..., B, C] (donde N >= 2 ) y con índices ordenados en el orden lexicográfico canónico.

Esta operación equivale a aplicar el tf.nn.softmax() normal a cada submatriz lógica más interna con forma [B, C] , pero con el inconveniente de que los elementos implícitamente cero no participan . Específicamente, el algoritmo es equivalente al siguiente:

(1) Aplica tf.nn.softmax() a una vista densificada de cada submatriz más interna con forma [B, C] , a lo largo de la dimensión de tamaño C; (2) Enmascara las ubicaciones originales implícitamente cero; (3) Renormaliza los elementos restantes.

Por lo tanto, el resultado SparseTensor tiene exactamente los mismos índices y forma distintos de cero.

Argumentos:

  • alcance: un objeto de alcance
  • índices_sp: 2-D. Matriz NNZ x R con los índices de valores no vacíos en un SparseTensor, en orden canónico.
  • sp_values: 1-D. Valores NNZ no vacíos correspondientes a sp_indices .
  • sp_shape: 1-D. Forma de la entrada SparseTensor.

Devoluciones:

  • Output : 1-D. Los valores NNZ para el resultado SparseTensor .

Constructores y destructores

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

Atributos públicos

operation
output

Funciones públicas

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

Atributos públicos

operación

Operation operation

producción

::tensorflow::Output output

Funciones públicas

EscasoSoftmax

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

nodo

::tensorflow::Node * node() const 

operador::tensorflow::Entrada

 operator::tensorflow::Input() const 

operador::tensorflow::Salida

 operator::tensorflow::Output() const