In #5268, we use t = torch.squeeze(t, dim=tuple(range(tensor_expand))) and it turns out that squeeze_copy.dims is not lowered. So we should look into this.

On the other hand, maybe @khatwanimohit can restore the loop change to workaround.

cc @wonjoolee95 to help @khatwanimohit to lower the op.