TensorDiag

TensorDiag kelas akhir publik

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

Metode Warisan

Konstanta

String akhir statis publik OP_NAME

Nama operasi ini dikenal dengan mesin inti TensorFlow

Nilai Konstan: "Diag"

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

Keluaran publik <T> keluaran ()