tensorflow :: operaciones :: SparseSoftmax

#include <sparse_ops.h>

Aplica softmax a un ND SparseTensor .

Resumen

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

Esta tf.nn.softmax() es equivalente a aplicar el tf.nn.softmax() normal a cada submatriz lógica más interna con forma [B, C] , pero con la trampa de que los elementos implícitamente cero no participan . Específicamente, el algoritmo es equivalente a lo 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 de SparseTensor tiene exactamente los mismos índices y forma distintos de cero.

Argumentos:

  • alcance: un objeto de alcance
  • sp_indices: 2-D. NNZ x R con los índices de valores no vacíos en un SparseTensor, en orden canónico.
  • sp_values: 1-D. NNZ valores 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 publicas

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 publicas

SparseSoftmax

 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