(self, t0, t1)
| 84 | return (a, b, 1) if a < b else (b, a, -1) |
| 85 | |
| 86 | def __call__(self, t0, t1): |
| 87 | t0, t1, sign = self.sort(t0, t1) |
| 88 | w = torch.stack([tree(t0, t1) for tree in self.trees]) * (self.sign * sign) |
| 89 | return w if self.batched else w[0] |
| 90 | |
| 91 | |
| 92 | class BrownianTreeNoiseSampler: |