(self, n, device='cpu', dtype=torch.float32)
| 385 | self.scale = scale |
| 386 | |
| 387 | def __call__(self, n, device='cpu', dtype=torch.float32): |
| 388 | return torch.sigmoid(self.loc + self.scale * torch.randn(n)).to(device).to(dtype) |
| 389 | |
| 390 | |
| 391 | """ Flow Model """ |
nothing calls this directly
no outgoing calls
no test coverage detected