tensorflow::ops::MatrixDiag

Verilen gruplandırılmış köşegen değerleriyle gruplandırılmış bir köşegen tensör döndürür.

Özet

Verilen bir diagonal için, bu işlem diagonal içeren ve geri kalan her şeyi sıfırlarla dolduran bir tensör döndürür. Köşegen şu şekilde hesaplanır:

diagonal k boyutlu [I, J, K, ..., N] olduğunu varsayarsak, çıktı [I, J, K, ..., N, N] boyutlarında k+1 mertebeli bir tensör olur; burada:

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

Örneğin:

# '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)

Argümanlar:

• kapsam: Bir Kapsam nesnesi
• köşegen: k derece, burada k >= 1 .

İadeler:

• Output : Sıra k+1 , output.shape = diagonal.shape + [diagonal.shape[-1]] .

Genel özellikler

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