tf.raw_ops.SparseSoftmax
Применяет softmax к пакетному N-D SparseTensor.
tf.raw_ops.SparseSoftmax(
sp_indices, sp_values, sp_shape, name=None
)
Входные данные представляют собой N-D SparseTensor с логической формой [..., B, C] (где N >= 2) и индексами, отсортированными в каноническом лексикографическом порядке.
Эта операция эквивалентна применению обычного tf.nn.softmax() к каждой самой внутренней логической подматрице формы [B, C], однако неявно нулевые элементы не участвуют. В частности, алгоритм эквивалентен следующему:
(1) Применяет tf.nn.softmax() к плотному представлению каждой самой внутренней подматрицы формы [B, C] вдоль измерения размера C; (2) Маскирует исходные неявно нулевые позиции; (3) Повторно нормализует оставшиеся элементы.
Таким образом, результат SparseTensor имеет в точности те же ненулевые индексы и форму.
| Аргументы | |
|---|---|
sp_indices | Tensor типа int64. 2-D. NNZ x R матрица с индексами непустых значений в SparseTensor в каноническом порядке. |
sp_values | Tensor. Должен иметь один из следующих типов: half, float32, float64. 1-D. NNZ непустых значений, соответствующих sp_indices. |
sp_shape | Tensor типа int64. 1-D. Форма входного SparseTensor. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
Tensor. Имеет тот же тип, что и sp_values. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/raw_ops/SparseSoftmax
