(self, grid_size: Sequence[int])
| 1987 | self.device = device |
| 1988 | |
| 1989 | def randomize(self, grid_size: Sequence[int]) -> None: |
| 1990 | self.random_offset = self.R.normal(size=([len(grid_size)] + list(grid_size))).astype(np.float32, copy=False) |
| 1991 | self.rand_mag = self.R.uniform(self.magnitude[0], self.magnitude[1]) |
| 1992 | |
| 1993 | def __call__(self, spatial_size: Sequence[int]) -> torch.Tensor: |
| 1994 | """ |