diffusion_models.mri_forward.undersampling_mask.naive_undersampling2d

diffusion_models.mri_forward.undersampling_mask.naive_undersampling2d(size, undersampling_ratio)[source]

2D regular subsampling with given undersampling ratio.

Parameters:

  • size (tensor) – shape tensor

  • undersampling_ratio (int) – integer representing the acceleration in both dimensions

Returns:

2D tensor with shape of last 2 dim of size

Return type:

Tensor