Hi.

I would like to know if there is a inplace version of the following series of operations on the tensor input: Basically, I have a tensor called input, and I would like to reshape and reindex along a specific dimension

The series of operations could be the following:

s=input.shape

s[1] = new_len

output = torch.zeros(s)

output[:,index_tensor,…]=input

but the last one takes lots of time, almost as much as a convolution. Any idea how to get something similar but way faster?

Thanks