Tensors have a "put_" method taking 2 arguments:
i.e.
t.put_(torch.tensor(list(mapping.keys()),dtype=torch.long),torch.tensor(list(mapping.values())))
speedup of about 10-15x over manual indexing. Graph of speedup