tensorflow::ops::MatrixSetDiag

Mengembalikan tensor matriks batch dengan nilai diagonal batch baru.

Ringkasan

Dengan diberikan input dan diagonal , operasi ini mengembalikan tensor dengan bentuk dan nilai yang sama seperti input , kecuali untuk diagonal utama dari matriks terdalam. Nilai-nilai ini akan ditimpa oleh nilai-nilai dalam diagonal .

Outputnya dihitung sebagai berikut:

Misalkan input memiliki k+1 dimensi [I, J, K, ..., M, N] dan diagonal memiliki k dimensi [I, J, K, ..., min(M, N)] . Maka outputnya adalah tensor berperingkat k+1 dengan dimensi [I, J, K, ..., M, N] di mana:

• output[i, j, k, ..., m, n] = diagonal[i, j, k, ..., n] untuk m == n .
• output[i, j, k, ..., m, n] = input[i, j, k, ..., m, n] untuk m != n .

Argumen:

• Cakupan: Objek Cakupan
• Masukan: Peringkat k+1 , di mana k >= 1 .
• Diagonal: Peringkat k , di mana k >= 1 .

Mengembalikan:

• Output : Rank k+1 , with output.shape = input.shape .

Atribut publik

Operation operation

::tensorflow::Output output

 MatrixSetDiag(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input diagonal
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const