Дан тензор булевой маски формы [Нет, # количество временных шагов] (Все установлено в false). Также есть еще один тензор с индексами, хранящимися для каждой партии. Диапазон индекса (0, # количество временных шагов). Его форма [Нет, 1]. Использование тензора индекса для каждого пакета в отдельности должно обновить срез исходной логической маски до True.
Например, считайте, что маска имеет вид [[False, False, False, False], [False, False] , False, False]], а тензор индекса должен быть [2,3]. Результат должен быть [[True, True, False, False], [True, True, True, False]]. т.е. в первом пакете [: 2] установлено значение true, а во втором пакете [: 3] установлено значение true.
Как достичь вышеуказанного в тензорном потоке без использования сеансовых вызовов и для всех пакетов сразу ?