tensor akışı:: işlem:: MatrixDiag

Belirli bir toplu diyagonal değerlere sahip toplu bir diyagonal tensör döndürür.

Özet

Bir diagonal verildiğinde, bu işlem diagonal bir tensör döndürür ve diğer her şey sıfırlarla doldurulur. Köşegen şu şekilde hesaplanır:

diagonal k boyutu olduğunu varsayalım [I, J, K, ..., N] , bu durumda çıktı k+1 dereceli ve [I, J, K, ..., N, N]` boyutlarına sahip 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
• diyagonal: Derece k , burada k >= 1 .

İade:

• Output : Rank k+1 , output.shape = diagonal.shape + [diagonal.shape[-1]] ile.

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