diffusion_models.mri_forward.undersampling_mask.gaussian_kernel1d_v

diffusion_models.mri_forward.undersampling_mask.gaussian_kernel1d_v(size, rel_sigma)[source]

1D Gaussian kernel repeated along vertical dimension.

Parameters:
  • size (Tensor) – shape tensor

  • rel_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