Mengembalikan tensor diagonal dengan nilai diagonal tertentu.
Jika diberi `diagonal`, operasi ini mengembalikan tensor dengan `diagonal` dan yang lainnya diisi dengan nol. Diagonal dihitung sebagai berikut:
Asumsikan `diagonal` memiliki dimensi [D1,..., Dk], maka outputnya adalah tensor dengan peringkat 2k dengan dimensi [D1,..., Dk, D1,..., Dk] dengan:
`output[i1,..., ik, i1,..., ik] = diagonal[i1, ..., ik]` dan 0 di tempat lain.
Misalnya:
# 'diagonal' is [1, 2, 3, 4]
tf.diag(diagonal) ==> [[1, 0, 0, 0]
[0, 2, 0, 0]
[0, 0, 3, 0]
[0, 0, 0, 4]]
Konstanta
| Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Publik
| Keluaran <T> | sebagai Keluaran () Mengembalikan pegangan simbolis tensor. |
| statis <T memperluas TType > TensorDiag <T> | |
| Keluaran <T> | keluaran () |
Metode Warisan
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Metode Publik
Keluaran publik <T> sebagai Keluaran ()
Mengembalikan pegangan simbolis tensor.
Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
TensorDiag statis publik <T> buat ( Lingkup lingkup , Operan <T> diagonal)
Metode pabrik untuk membuat kelas yang membungkus operasi TensorDiag baru.
Parameter
| cakupan | ruang lingkup saat ini |
|---|---|
| diagonal | Peringkat k tensor dimana k paling banyak 1. |
Kembali
- contoh baru TensorDiag