(self, device, seeds)
| 181 | |
| 182 | class StackedRandomGenerator: |
| 183 | def __init__(self, device, seeds): |
| 184 | super().__init__() |
| 185 | self.generators = [torch.Generator(device).manual_seed(int(seed) % (1 << 32)) for seed in seeds] |
| 186 | |
| 187 | def randn(self, size, **kwargs): |
| 188 | assert size[0] == len(self.generators) |
nothing calls this directly
no outgoing calls
no test coverage detected