tensorflow::ops::SparseSoftmax

#include <sparse_ops.h>

Применяет softmax к пакетному N-мерному SparseTensor.

Краткое описание

Входные данные представляют собой N-мерный SparseTensor с логической формой [..., B, C] (где N >= 2), а его индексы отсортированы в каноническом лексикографическом порядке.

Эта операция эквивалентна применению обычного tf.nn.softmax() к каждой внутренней логической подматрице формы [B, C], но с тем отличием, что неявно нулевые элементы не участвуют. В частности, алгоритм эквивалентен следующему:

(1) Применяет tf.nn.softmax() к плотному представлению каждой внутренней подматрицы формы [B, C] вдоль измерения размера C; (2) Маскирует исходные позиции неявных нулей; (3) Повторно нормализует оставшиеся элементы.

Таким образом, результат SparseTensor имеет в точности те же ненулевые индексы и форму.

Аргументы:

• scope: объект Scope
• sp_indices: двумерная матрица NNZ x R с индексами непустых значений в SparseTensor в каноническом порядке.
• sp_values: одномерные NNZ непустые значения, соответствующие sp_indices.
• sp_shape: одномерная форма входного SparseTensor.

Возвращает:

• Output: одномерные значения NNZ для результата SparseTensor.

Открытые атрибуты

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