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

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

#include <array_ops.h>

दिए गए विकर्ण मानों के साथ एक बैचित विकर्ण तन्यता लौटाता है।

सारांश

एक diagonal को देखते हुए, यह ऑपरेशन diagonal साथ एक तानवाला देता है और बाकी सब शून्य के साथ गद्देदार होता है। विकर्ण की गणना निम्नानुसार की जाती है:

मान लें कि diagonal के k आयाम हैं [I, J, K, ..., N] , तो आउटपुट आयामों [I, J, K, ..., N, N] `के साथ रैंक k+1 का दसवां भाग है:

output[i, j, k, ..., m, n] = 1{m=n} * diagonal[i, j, k, ..., n]

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

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

and diagonal.shape = (2, 4)

tf.matrix_diag(diagonal) ==> [[[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]]]

which has shape (2, 4, 4)

तर्क:

  • गुंजाइश: एक स्कोप ऑब्जेक्ट
  • विकर्ण: रैंक k , जहां k >= 1

रिटर्न:

  • Output : रैंक k+1 , Output साथ। output.shape = diagonal.shape + [diagonal.shape[-1]]

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

MatrixDiag (const :: tensorflow::Scope & scope, :: tensorflow::Input diagonal)

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

operation
output

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

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

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

ऑपरेशन

Operation operation

उत्पादन

::tensorflow::Output output

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

मैट्रिक्सडायग

 MatrixDiag(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input diagonal
)

नोड

::tensorflow::Node * node() const 

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

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

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

 operator::tensorflow::Output() const