diffusion_models.mri_forward.undersampling_mask.UndersamplingMask

class diffusion_models.mri_forward.undersampling_mask.UndersamplingMask(mask_type, undersampling_ratio, device=None)[source]

Bases: Module

__init__(mask_type, undersampling_ratio, device=None)[source]

Methods

__init__(mask_type, undersampling_ratio[, ...])

forward(x)