Mengembalikan tensor satu-panas.
Lokasi yang diwakili oleh indeks dalam `indeks` mengambil nilai `on_value`, sedangkan semua lokasi lainnya mengambil nilai `off_value`.
Jika `indeks` masukan adalah peringkat `N`, output akan memiliki peringkat `N+1`, Sumbu baru dibuat pada dimensi `sumbu` (default: sumbu baru ditambahkan di akhir).
Jika `indeks` adalah skalar, bentuk keluarannya akan berupa vektor dengan panjang `kedalaman`.
Jika `indeks` adalah vektor dengan panjang `fitur`, bentuk keluarannya adalah:
features x depth if axis == -1
depth x features if axis == 0
batch x features x depth if axis == -1
batch x depth x features if axis == 1
depth x batch x features if axis == 0
Misalkan itu
indices = [0, 2, -1, 1]
depth = 3
on_value = 5.0
off_value = 0.0
axis = -1
output =
[5.0 0.0 0.0] // one_hot(0)
[0.0 0.0 5.0] // one_hot(2)
[0.0 0.0 0.0] // one_hot(-1)
[0.0 5.0 0.0] // one_hot(1)
indices = [0, 2, -1, 1]
depth = 3
on_value = 0.0
off_value = 3.0
axis = 0
output =
[0.0 3.0 3.0 3.0]
[3.0 3.0 3.0 0.0]
[3.0 3.0 3.0 3.0]
[3.0 0.0 3.0 3.0]
// ^ one_hot(0)
// ^ one_hot(2)
// ^ one_hot(-1)
// ^ one_hot(1)
indices = [[0, 2], [1, -1]]
depth = 3
on_value = 1.0
off_value = 0.0
axis = -1
output =
[
[1.0, 0.0, 0.0] // one_hot(0)
[0.0, 0.0, 1.0] // one_hot(2)
][
[0.0, 1.0, 0.0] // one_hot(1)
[0.0, 0.0, 0.0] // one_hot(-1)
]
Kelas Bersarang
kelas | OneHot.Opsi | Atribut opsional untuk OneHot |
Metode Publik
Keluaran <U> | sebagai Keluaran () Mengembalikan pegangan simbolik tensor. |
OneHot.Options statis | sumbu (Sumbu panjang) |
statis <U, T memperluas Nomor> OneHot <U> | |
Keluaran <U> | keluaran () Tensor satu-panas. |
Metode Warisan
Metode Publik
Keluaran publik <U> sebagai Keluaran ()
Mengembalikan pegangan simbolik tensor.
Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
Sumbu OneHot.Options statis publik (Sumbu panjang)
Parameter
sumbu | Sumbu yang akan diisi (default: -1, sumbu paling dalam yang baru). |
---|
public static OneHot <U> buat ( Lingkup cakupan , indeks Operand <T>, kedalaman Operand <Integer>, Operand <U> onValue, Operand <U> offValue, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi OneHot baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
indeks | Tensor indeks. |
kedalaman | Skalar yang menentukan kedalaman satu dimensi panas. |
padaNilai | Skalar yang menentukan nilai untuk mengisi keluaran ketika `indeks[j] = i`. |
nilai mati | Skalar yang menentukan nilai untuk mengisi keluaran ketika `indeks[j] != i`. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru OneHot