tf.raw_ops.SparseSoftmax

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

tf.raw_ops.SparseSoftmax(
    sp_indices, sp_values, sp_shape, name=None
)

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

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

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

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