टेंसरफ़्लो:: ऑप्स:: मैट्रिक्सडिआगपार्ट

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

सारांश

यह ऑपरेशन बैच किए गए 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:])] ।

सार्वजनिक गुण

::tensorflow::Output diagonal

Operation operation

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const