OneHot

یک تانسور یک داغ را برمی‌گرداند.

مکان‌های نشان‌داده‌شده توسط شاخص‌ها در «شاخص‌ها» مقدار «روی_مقدار» را می‌گیرند، در حالی که همه مکان‌های دیگر مقدار «خارج_مقدار» را می‌گیرند.

اگر «شاخص‌های» ورودی رتبه «N» باشد، خروجی دارای رتبه «N+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 × 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)
   ]