tensorflow :: operaciones :: SparseSoftmax

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 el 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 .

Atributos públicos

Operation operation

::tensorflow::Output output

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const