Получить маску из индекса (top_k) в тензорном потоке
У меня есть два тензора, a и b, a - тензор top-k, b - тензор маски. форма a - это [batch_size, k], форма b - [batch_size * seq_len], dtype - это bool, все инициализируется как False. каждая строка a имеет k целых чисел, каждое целочисленное значение i означает, что i-й элемент соответствующего необработанного числа b должен быть установлен в True.
Например: b [[Ложь, Ложь, Ложь, Ложь, Ложь], [Ложь, Ложь, Ложь, Ложь, Ложь]].
a равно [[0,4],[1,2]], установите соответствующий индекс a в b равным True. тогда результат будет [[True, False, False, False, True], [False, True, True, False, False]].