PyTorch Forums
How can we convert pykeops lazytensor back to a torch tensor in the forward layer of a model?
torchlight
July 26, 2020, 10:06pm
1
https://www.kernel-operations.io/keops/python/api/pytorch/LazyTensor.html#pykeops.torch.LazyTensor