(self, grid_size: Sequence[int])
| 2929 | self.device = device |
| 2930 | |
| 2931 | def randomize(self, grid_size: Sequence[int]) -> None: |
| 2932 | super().randomize(None) |
| 2933 | if not self._do_transform: |
| 2934 | return None |
| 2935 | self.rand_offset = self.R.uniform(-1.0, 1.0, [3] + list(grid_size)).astype(np.float32, copy=False) |
| 2936 | self.magnitude = self.R.uniform(self.magnitude_range[0], self.magnitude_range[1]) |
| 2937 | self.sigma = self.R.uniform(self.sigma_range[0], self.sigma_range[1]) |
| 2938 | self.rand_affine_grid.randomize() |
| 2939 | |
| 2940 | def __call__( |
| 2941 | self, |