diffusion_models.mri_forward.undersampling_mask.StochasticUndersamplingMask

class diffusion_models.mri_forward.undersampling_mask.StochasticUndersamplingMask(mask_type, rel_sigma, undersampling_rate, random_seed=42, device=None)[source]

Bases: Module

__init__(mask_type, rel_sigma, undersampling_rate, random_seed=42, device=None)[source]

Methods

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

forward(x)