TensorDiagPart kelas akhir publik
Mengembalikan bagian diagonal tensor.
Operasi ini mengembalikan tensor dengan bagian `diagonal` dari `input`. Bagian `diagonal` dihitung sebagai berikut:
Asumsikan `input` berdimensi `[D1,..., Dk, D1,..., Dk]`, maka outputnya berupa tensor rank `k` dengan dimensi `[D1,..., Dk]` Di mana:
`diagonal[i1,..., ik] = masukan[i1, ..., ik, i1,..., ik]`.
Misalnya:
# 'input' is [[1, 0, 0, 0]
[0, 2, 0, 0]
[0, 0, 3, 0]
[0, 0, 0, 4]]
tf.diag_part(input) ==> [1, 2, 3, 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 > TensorDiagPart <T> | |
| Keluaran <T> | diagonal () Diagonal yang diekstraksi. |
Metode Warisan
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Nilai Konstan: "DiagPart"
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.
public static TensorDiagPart <T> buat ( Lingkup lingkup , masukan Operan <T>)
Metode pabrik untuk membuat kelas yang membungkus operasi TensorDiagPart baru.
Parameter
| cakupan | ruang lingkup saat ini |
|---|---|
| masukan | Beri peringkat k tensor dengan k genap dan bukan nol. |
Kembali
- contoh baru TensorDiagPart