Mengembalikan tensor satu-panas.
Lokasi yang diwakili oleh indeks di `indices` mengambil nilai `on_value`, sementara semua lokasi lain mengambil nilai `off_value`.
Jika `indeks` input 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 adalah vektor dengan panjang `kedalaman`.
Jika `indices` adalah vektor dengan panjang` features`, bentuk output akan:
features x depth if axis == -1
depth x features if axis == 0
jika `indices` adalah matriks (batch) dengan bentuk` [batch, fitur] `, bentuk output akan: batch x features x depth if axis == -1
batch x depth x features if axis == 1
depth x batch x features if axis == 0
Contoh =========Misalkan
indices = [0, 2, -1, 1]
depth = 3
on_value = 5.0
off_value = 0.0
axis = -1
Kemudian output `[4 x 3]`: 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)
Misalkan indices = [0, 2, -1, 1]
depth = 3
on_value = 0.0
off_value = 3.0
axis = 0
Kemudian output `[3 x 4]`: 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)
Misalkan indices = [[0, 2], [1, -1]]
depth = 3
on_value = 1.0
off_value = 0.0
axis = -1
Kemudian output `[2 x 2 x 3] `: 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
Output <U> | asOutput () Mengembalikan pegangan simbolis tensor. |
statis OneHot.Options | axis (Long axis) |
statis <U, T meluas Nomor> onehot <U> | |
Output <U> | Output () Tensor satu panas. |
Metode yang Diwarisi
Metode Publik
publik Keluaran <U> asOutput ()
Mengembalikan pegangan simbolis tensor.
Input ke operasi TensorFlow adalah output dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
public static OneHot.Options sumbu (Long axis)
Parameter
sumbu | Sumbu yang akan diisi (default: -1, sumbu paling dalam baru). |
---|
public static onehot <U> membuat ( Lingkup lingkup, Operan <T> indeks, Operan <Integer> mendalam, Operan <U> onValue, Operan <U> offValue, Options ... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi OneHot baru.
Parameter
cakupan | lingkup saat ini |
---|---|
indeks | Sebuah tensor indeks. |
kedalaman | Sebuah skalar mendefinisikan kedalaman satu dimensi panas. |
padaNilai | Sebuah skalar mendefinisikan nilai untuk mengisi output ketika `indeks[j] = i`. |
offNilai | Sebuah skalar mendefinisikan nilai untuk mengisi output ketika `indeks[j] != i`. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru OneHot