TensorDiagPart

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>
buat ( Lingkup lingkup , masukan Operan <T>)
Metode pabrik untuk membuat kelas yang membungkus operasi TensorDiagPart baru.
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

Keluaran publik <T> diagonal ()

Diagonal yang diekstraksi.