tensorflow::ops::SparseSoftmax
#include <sparse_ops.h> Применяет функцию softmax к пакетному разреженному SparseTensor ND.
Краткое содержание
Входные данные представляют собой разреженный тензор ND с логической формой [..., B, C] (где N >= 2 ) и индексами, отсортированными в каноническом лексикографическом порядке.
Эта операция эквивалентна применению обычной функции tf.nn.softmax() к каждой внутренней логической подматрице формы [B, C] , но с той оговоркой, что неявно нулевые элементы не участвуют в процессе . В частности, алгоритм эквивалентен следующему:
(1) Применяет tf.nn.softmax() к уплотненному представлению каждой внутренней подматрицы формы [B, C] вдоль измерения размера C; (2) Маскирует исходные неявно нулевые местоположения; (3) Перенормирует оставшиеся элементы.
Следовательно, результат SparseTensor имеет точно такие же ненулевые индексы и форму.
Аргументы:
- scope: Объект Scope
- sp_indices: двумерная матрица
NNZ x R, содержащая индексы непустых значений в разреженном тензоре в каноническом порядке. - sp_values: 1-мерные непустые значения
NNZ, соответствующиеsp_indices. - sp_shape: 1-мерная форма входного разреженного тензора.
Возвращает:
-
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
выход
::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::Input
operator::tensorflow::Input() const
оператор::tensorflow::Output
operator::tensorflow::Output() const