diffusion_models.models.multicoil.MultiCoilConv2d

class diffusion_models.models.multicoil.MultiCoilConv2d(*args, **kwargs)[source]

Bases: Module

__init__(*args, **kwargs)[source]

Methods

__init__(*args, **kwargs)

forward(x)

rtype:

Tensor, 'batch coils out_channels height width')