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: 2-мерная.
NNZ x Rматрица с индексами ненулевых значений в SparseTensor, в каноническом порядке. - sp_values: 1-мерная.
NNZненулевые значения, соответствующиеsp_indices. - sp_shape: 1-мерная. Форма входного SparseTensor.
Возвращает:
-
Output: 1-мерная.NNZзначения для результатаSparseTensor.
| Конструкторы и деструкторы | |
|---|---|
SparseSoftmax(const ::tensorflow::Scope & scope, ::tensorflow::Input sp_indices, ::tensorflow::Input sp_values, ::tensorflow::Input sp_shape) |
| Публичные атрибуты | |
|---|---|
operation | |
output | |
| Публичные функции | |
|---|---|
node() const | ::tensorflow::Node * |
operator::tensorflow::Input() const | |
operator::tensorflow::Output() const | |
Публичные атрибуты
operation
Operation operation
вывод
::tensorflow::Output output
Публичные функции
SparseSoftmax
SparseSoftmax( const ::tensorflow::Scope & scope, ::tensorflow::Input sp_indices, ::tensorflow::Input sp_values, ::tensorflow::Input sp_shape )
узел
::tensorflow::Node * node() const
оператор::tensorflow::Вход
operator::tensorflow::Input() const
оператор::tensorflow::Вывод
operator::tensorflow::Output() const
© 2020 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/versions/r2.4/api_docs/cc/class/tensorflow/ops/sparse-softmax