OneHot

एक-गर्म टेंसर लौटाता है।

`सूचकांक` में सूचकांकों द्वारा दर्शाए गए स्थान `on_value` मान लेते हैं, जबकि अन्य सभी स्थान `off_value` मान लेते हैं।

यदि इनपुट `इंडेक्स` रैंक `एन` है, तो आउटपुट में रैंक `एन+1` होगा, नया अक्ष आयाम `अक्ष` पर बनाया गया है (डिफ़ॉल्ट: नई धुरी अंत में संलग्न है)।

यदि `सूचकांक` एक अदिश है तो आउटपुट आकार `गहराई` लंबाई का सदिश होगा।

यदि `indices` लंबाई` features` का एक वेक्टर है, उत्पादन आकार होगा:

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

यदि `indices` एक मैट्रिक्स आकार के साथ (बैच)` है [बैच, सुविधाओं] `, उत्पादन आकार होगा:

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 एक्स 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 एक्स 2 एक्स 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)
   ]