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 имеет ровно те же ненулевые индексы и форму.