тензорный поток:: опс:: РазреженныйSoftmax

#include <sparse_ops.h>

Применяет softmax к пакетному ND SparseTensor .

Краткое содержание

Входные данные представляют собой ND SparseTensor с логической формой [..., B, C] (где N >= 2 ) и с индексами, отсортированными в каноническом лексикографическом порядке.

Эта операция эквивалентна применению обычного tf.nn.softmax() к каждой самой внутренней логической подматрице с формой [B, C] , но с той загвоздкой, что неявно нулевые элементы не участвуют . В частности, алгоритм эквивалентен следующему:

(1) Применяет tf.nn.softmax() к уплотненному представлению каждой самой внутренней подматрицы формы [B, C] по размеру размера C; (2) Маскирует исходные неявно нулевые местоположения; (3) Перенормирует оставшиеся элементы.

Следовательно, результат SparseTensor имеет точно такие же ненулевые индексы и форму.

Аргументы:

  • область: объект области .
  • sp_indices: 2-D. Матрица NNZ x R с индексами непустых значений в SparseTensor в каноническом порядке.
  • sp_values: 1-D. Непустые значения NNZ , соответствующие sp_indices .
  • sp_shape: 1-D. Форма входного SparseTensor.

Возврат:

  • Output : 1-D. Значения 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

Общественные функции

РазреженныйSoftmax

 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::Выход

 operator::tensorflow::Output() const