📚 Documentation

https://pytorch.org/docs/master/torch.html#torch.lerp

The shapes of start and end must be broadcastable. If weight is a tensor, then the shapes of start, end must be broadcastable.

Should probably be weight, start, end must be broadcastable at the end.