I have one_tensor (3,1) of values, say:

one_tensor = torch.tensor([[24],[36],[12]])

and index_tensor (3,1) of indices, say:

index_tensor = torch.tensor([[0],[3],[2]])

I’d like to map the values of one_tensor to a third_tensor (3,4) of zeros:

third_tensor = torch.zeros(3,4)

using the indices of index_tensor so that third tensor ends up like this:

third_tensor = tensor([[24, 0, 0, 0], [0, 0, 0, 36], [0, 0, 12, 0]])

Is there a fast, efficient way to do this without manually iterating over each row?