diffusion_models.mri_forward.undersampling_mask.gaussian_kernel2d¶
- diffusion_models.mri_forward.undersampling_mask.gaussian_kernel2d(size, rel_sigma)[source]¶
Gaussian kernel as 2D tensor.
- Parameters:
size (
Tensor) – shape tensorrel_sigma (
float) – std dev of kernel, if image width/height was 1
- Returns:
2D tensor with shape of last 2 dim of size
- Return type:
Tensor