How can we convert pykeops lazytensor back to a torch tensor in the forward layer of a model?

https://www.kernel-operations.io/keops/python/api/pytorch/LazyTensor.html#pykeops.torch.LazyTensor