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
output
::tensorflow::Output output
Общедоступные функции
SparseSoftmax
SparseSoftmax( const ::tensorflow::Scope & scope, ::tensorflow::Input sp_indices, ::tensorflow::Input sp_values, ::tensorflow::Input sp_shape )
node
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/cc/class/tensorflow/ops/sparse-softmax