टेंसरफ़्लो :: ऑप्स :: 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