सहायता Kaggle पर TensorFlow साथ ग्रेट बैरियर रीफ की रक्षा चैलेंज में शामिल हों

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

#include <array_ops.h>

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

सारांश

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

मान लें कि input k आयाम हैं [I, J, K, ..., M, N] , तो आउटपुट रैंक k - 1 का एक आयाम है जिसमें आयाम [I, J, K, ..., min(M, N)] कहां:

diagonal[i, j, k, ..., n] = input[i, j, k, ..., n, n]

इनपुट कम से कम एक मैट्रिक्स होना चाहिए।

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

# 'input' is [[[1, 0, 0, 0]
               [0, 2, 0, 0]
               [0, 0, 3, 0]
               [0, 0, 0, 4]],
              [[5, 0, 0, 0]
               [0, 6, 0, 0]
               [0, 0, 7, 0]
               [0, 0, 0, 8]]]

and input.shape = (2, 4, 4)

tf.matrix_diag_part(input) ==> [[1, 2, 3, 4], [5, 6, 7, 8]]

which has shape (2, 4)

तर्क:

  • गुंजाइश: एक स्कोप ऑब्जेक्ट
  • इनपुट: रैंक k टेंसर जहां k >= 2

रिटर्न:

  • Output : निकाले गए विकर्ण (ओं) का आकार diagonal.shape = input.shape[:-2] + [min(input.shape[-2:])]

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

MatrixDiagPart (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

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

MatrixDiagPart

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

नोड

::tensorflow::Node * node() const 

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

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

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

 operator::tensorflow::Output() const