Solved by ptrblck in post #4 
              
              
                There are some minor error regarding the shapes. 
Here is a working solution for batched inputs: 
def slerp(val, low, high):
    low_norm = low/torch.norm(low, dim=1, keepdim=True)
    high_norm = high/torch.norm(high, dim=1, keepdim=True)
    omega = torch.acos((low_norm*high_norm).sum(1))
    so =…