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)