(self, sigma, sigma_next)
| 110 | self.tree = BatchedBrownianTree(x, t0, t1, seed) |
| 111 | |
| 112 | def __call__(self, sigma, sigma_next): |
| 113 | t0, t1 = self.transform(torch.as_tensor(sigma)), self.transform(torch.as_tensor(sigma_next)) |
| 114 | return self.tree(t0, t1) / (t1 - t0).abs().sqrt() |
| 115 | |
| 116 | |
| 117 | @torch.no_grad() |
nothing calls this directly
no outgoing calls
no test coverage detected