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