у меня есть 4 тензора
ib0 = torch.tensor([14,17])
bb0 = torch.tensor([0,1])
ib1 = torch.tensor([17,4,2])
bb1 = torch.tensor([0,0,1])
мне нужно получить другой тензор, который соответствует значениям в ib0 и ib1 на основе значений bb0 и bb1. результат, который я ищу, равен
torch.tensor([[14,17],[14,4],[17,2]])
, поскольку индекс 0 (из bb0) в ib0 имеет значение 14, а индексы 0 (из bb1) имеют значения 17,4 в ib1 и аналогичны для индекс 1 в bb0. Это необходимо в алгоритме поиска луча для объединения индексов из одного луча. Тензоры bb0 и bb1 - это индексы лучей, в то время как ib0, ib1 - это индексы в 2 лучах 0 и 1. Таким образом, описанная выше процедура является просто способом убедиться, что индексы из разных лучей не сливаются.