Карту каждого элемента Torch.tensor с его значением в DICT ⇐ Python

[Anonymous]

Предположим, что у меня есть тензор T , состоящий только в нулевых значениях:

Код: Выделить всё

t = torch.Tensor([1, 0, 0, 1])

и дикта с весами :

Код: Выделить всё

weights = {0: 0.1, 1: 0.9}

Я хочу сформировать новый тензор new_t , так что каждый элемент в Tensor T сопоставлен с соответствующим значением в весах DICT < /code>: < /p>

Код: Выделить всё

new_t = torch.Tensor([0.9, 0.1, 0.1, 0.9])

Есть ли элегантный способ сделать это без итерации над тензором t ? Я слышал о torch.apply , но это работает только в том случае, если Tensor T находится на процессоре, есть ли другие параметры?

Подробнее здесь: https://stackoverflow.com/questions/794 ... n-the-dict