Is your feature request related to a problem? Please describe.
when do_transform is False and the spatial_size isn't set, the resampling step of RandAffine could be skipped for better performance, for example:
|
if self._do_transform: |
|
grid = self.rand_affine_grid(spatial_size=sp_size) |
|
else: |
|
grid = create_grid(spatial_size=sp_size) |