def lookup(self, readings, level):
if self._need_repacking or self._mu.device != readings.device:
self.repack(readings.device)
if isinstance(level, torch.Tensor):
mu = torch_radon_cuda.readings_lookup_multilevel(
readings, self._mu, level)
sigma = torch_radon_cuda.readings_lookup_multilevel(
readings, self._sigma, level)
else:
mu = torch_radon_cuda.readings_lookup(readings, self._mu[level])
sigma = torch_radon_cuda.readings_lookup(readings,
self._sigma[level])
return mu, sigma
def readings_lookup(self, sensor_readings, lookup_table):
return torch_radon_cuda.readings_lookup(sensor_readings, lookup_table)