ML Topluluk Günü 9 Kasım! TensorFlow, JAX güncellemeler için bize katılın ve daha fazla bilgi edinin

tensorflow :: ops :: SparseSoftmax

#include <sparse_ops.h>

Bir toplu ND SparseTensor .

Özet

Girişler mantıksal şekle [..., B, C] (burada N >= 2 ) ve kanonik sözlük sırasına göre sıralanmış indekslere sahip bir ND SparseTensoru temsil eder.

Bu işlem, normal tf.nn.softmax() ı [B, C] , ancak örtük olarak sıfır elemanların katılmadığı yakalama ile en içteki mantıksal alt tf.nn.softmax() uygulamaya eşdeğerdir. Özellikle, algoritma aşağıdakilere eşdeğerdir:

(1) tf.nn.softmax() biçimini [B, C] ile C boyutu boyunca en içteki alt matrisin yoğunlaştırılmış görünümüne uygular; (2) Orijinal örtük olarak sıfır konumları maskeler; (3) Kalan öğeleri yeniden normalleştirir.

Bu nedenle, SparseTensor sonucu tamamen aynı sıfır olmayan indislere ve şekle sahiptir.

Argümanlar:

  • kapsam: Bir Scope nesnesi
  • sp_indices: 2-D. Bir SparseTensor'daki boş olmayan değerlerin endekslerini kanonik sırayla içeren NNZ x R matrisi.
  • sp_values: 1-D. NNZ karşılık gelen sp_indices boş olmayan değerler.
  • sp_shape: 1-D. Giriş SparseTensor'un şekli.

İadeler:

  • Output : 1-D. NNZ sonucunun SparseTensor değerleri.

Yapıcılar ve Yıkıcılar

SparseSoftmax (const :: tensorflow::Scope & scope, :: tensorflow::Input sp_indices, :: tensorflow::Input sp_values, :: tensorflow::Input sp_shape)

Genel özellikler

operation
output

Kamusal işlevler

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Genel özellikler

operasyon

Operation operation

çıktı

::tensorflow::Output output

Kamusal işlevler

SparseSoftmax

 SparseSoftmax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input sp_indices,
  ::tensorflow::Input sp_values,
  ::tensorflow::Input sp_shape
)

düğüm

::tensorflow::Node * node() const 

operator :: tensorflow :: Girdi

 operator::tensorflow::Input() const 

operator :: tensorflow :: Çıktı

 operator::tensorflow::Output() const