Google I/O एक लपेट है! TensorFlow सत्रों पर पकड़ बनाएं सत्र देखें

टेंसरफ़्लो :: ऑप्स :: डायगपार्ट

#include <array_ops.h>

टेंसर के विकर्ण भाग को लौटाता है।

सारांश

यह ऑपरेशन input के diagonal भाग के साथ एक टेंसर देता है। diagonal भाग की गणना निम्नानुसार की जाती है:

मान लें कि input के आयाम हैं [D1,..., Dk, D1,..., Dk] , तो आउटपुट आयाम [D1,..., Dk] साथ रैंक k [D1,..., Dk] :

diagonal[i1,..., ik] = input[i1, ..., ik, i1,..., ik]

उदाहरण के लिए:

# 'input' is [[1, 0, 0, 0]
              [0, 2, 0, 0]
              [0, 0, 3, 0]
              [0, 0, 0, 4]]

tf.diag_part(input) ==> [1, 2, 3, 4]

तर्क:

  • गुंजाइश: एक स्कोप ऑब्जेक्ट
  • इनपुट: रैंक k टेंसर जहां k सम है और शून्य नहीं है।

रिटर्न:

  • Output : निकाले गए विकर्ण।

कंस्ट्रक्टर और डिस्ट्रक्टर्स

DiagPart (const :: tensorflow::Scope & scope, :: tensorflow::Input input)

सार्वजनिक विशेषताएँ

diagonal
operation

सार्वजनिक कार्य

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

सार्वजनिक विशेषताएँ

विकर्ण

::tensorflow::Output diagonal

ऑपरेशन

Operation operation

सार्वजनिक कार्य

डायगपार्ट

 DiagPart(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input
)

नोड

::tensorflow::Node * node() const 

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

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

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

 operator::tensorflow::Output() const