У меня есть два тензора, 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]].