ワンホット

ワンホットテンソルを返します。

`indices` 内のインデックスによって表される位置は値 `on_value` をとり、他のすべての位置は値 `off_value` をとります。

入力「インデックス」がランク「N」の場合、出力はランク「N+1」になります。新しい軸はディメンション「axis」に作成されます (デフォルト: 新しい軸は最後に追加されます)。

`indices` がスカラーの場合、出力形状は長さ ` Depth` のベクトルになります。

`indices` が長さ `features` のベクトルの場合、出力形状は次のようになります。

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

`indices` が `[batch, features]` 形状を持つ行列 (バッチ) の場合、出力形状は次のようになります。

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

例 =========

次のように仮定します。

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

出力は `[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)
 

次のように仮定します。

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

出力は `[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)
 

次のように仮定します。

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

出力は `[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)
   ]