OneHot

Trả về một tenxơ một nóng.

Các vị trí được biểu thị bằng chỉ mục trong `chỉ mục` nhận giá trị `on_value`, trong khi tất cả các vị trí khác nhận giá trị `off_value`.

Nếu `chỉ số` đầu vào có thứ hạng `N` thì đầu ra sẽ có thứ hạng `N+1`, Trục mới được tạo ở thứ nguyên `trục` (mặc định: trục mới được thêm vào cuối).

Nếu `chỉ số` là vô hướng thì hình dạng đầu ra sẽ là vectơ có độ dài `độ sâu`.

Nếu `chỉ số` là một vectơ có độ dài `features`, hình dạng đầu ra sẽ là:

features x depth if axis == -1
   depth x features if axis == 0
 

Nếu `indices` là một ma trận (lô) có hình dạng `[batch, feature]`, hình dạng đầu ra sẽ là:

batch x features x depth if axis == -1
   batch x depth x features if axis == 1
   depth x batch x features if axis == 0
 

Ví dụ ==========

Giả sử rằng

indices = [0, 2, -1, 1]
   depth = 3
   on_value = 5.0
   off_value = 0.0
   axis = -1
 

Sau đó đầu ra là `[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)
 

Giả sử rằng

indices = [0, 2, -1, 1]
   depth = 3
   on_value = 0.0
   off_value = 3.0
   axis = 0
 

Sau đó đầu ra là `[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)
 

Giả sử rằng

indices = [[0, 2], [1, -1]]
   depth = 3
   on_value = 1.0
   off_value = 0.0
   axis = -1
 

Sau đó đầu ra là `[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)
   ]