I have a tensor with shape [1,100,34] which needs to be added with 2 tensors [1,100,17] shape at alternative indexes. Current Implementation is like this in pytorch.