Индекс Питорча с Тензором ⇐ Python
-
Гость
Индекс Питорча с Тензором
У меня есть двумерный тензор arr со всеми элементами, равными 0. У меня есть второй тензор idx. Я хочу сделать все записи в arr с индексами в idx равными 1.
arr = torch.zeros(size = (2,10)) idx = torch.Tensor([ [0,2], [4,5] ]) arr[idx] = 1 #Это не работает печать (обр.) Вывод должен выглядеть следующим образом:
tensor([[1., 0., 1., 0., 0., 0., 0., 0., 0., 0.], [0., 0., 0., 0., 1., 1., 0., 0., 0., 0.]]) Я был уверен, что обязательно найду кого-нибудь еще, кто задаст этот вопрос в SO, однако не смог его найти. Надеюсь, это не дубликат.
У меня есть двумерный тензор arr со всеми элементами, равными 0. У меня есть второй тензор idx. Я хочу сделать все записи в arr с индексами в idx равными 1.
arr = torch.zeros(size = (2,10)) idx = torch.Tensor([ [0,2], [4,5] ]) arr[idx] = 1 #Это не работает печать (обр.) Вывод должен выглядеть следующим образом:
tensor([[1., 0., 1., 0., 0., 0., 0., 0., 0., 0.], [0., 0., 0., 0., 1., 1., 0., 0., 0., 0.]]) Я был уверен, что обязательно найду кого-нибудь еще, кто задаст этот вопрос в SO, однако не смог его найти. Надеюсь, это не дубликат.
