Google I/O में ट्यूनिंग के लिए धन्यवाद। मांग पर सभी सत्र देखें मांग पर देखें

टेंसरफ़्लो :: ऑप्स :: SparseSoftmax

#include <sparse_ops.h>

एक batched एन डी करने के लिए लागू होता है softmax SparseTensor

सारांश

इनपुट तार्किक आकार [..., B, C] (जहाँ N >= 2 ) के साथ ND SparseTensor का प्रतिनिधित्व करते हैं, और कैनोनिक लिपिकोग्राफ़िक क्रम में छांटे गए सूचकांकों के साथ।

यह ऑप आकार [B, C] साथ प्रत्येक अंतरतम तार्किक tf.nn.softmax() में सामान्य tf.nn.softmax() को लागू करने के बराबर है, लेकिन इस पकड़ के साथ कि निहित शून्य तत्व भाग नहीं लेते हैं । विशेष रूप से, एल्गोरिथ्म निम्नलिखित के बराबर है:

(1) आकार-सी आयाम के साथ आकार [B, C] साथ प्रत्येक अंतरतम tf.nn.softmax() घनीभूत दृश्य के लिए tf.nn.softmax() पर लागू होता है; (2) मूल अंतर्निहित शून्य स्थानों को मास्क करता है; (३) शेष तत्वों का नवीनीकरण करता है।

इसलिए, SparseTensor परिणाम में बिल्कुल गैर-शून्य सूचक और आकृति होती है।

तर्क:

  • गुंजाइश: एक स्कोप ऑब्जेक्ट
  • sp_indices: 2-डी। NNZ x R मैट्रिक्स NNZ x R में गैर-रिक्त मानों के सूचकांकों के साथ, विहित क्रम में।
  • sp_values: 1-डी। NNZ गैर-रिक्त मान जो sp_indices अनुरूप हैं।
  • sp_shape: 1-D। इनपुट का आकार 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

उत्पादन

::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 

ऑपरेटर :: टेंसरफ़्लो :: इनपुट

 operator::tensorflow::Input() const 
है

ऑपरेटर :: टेंसरफ़्लो :: आउटपुट

 operator::tensorflow::Output() const