medil.vae.SparseLinear

class medil.vae.SparseLinear(*args: Any, **kwargs: Any)[source]

Bases: Module

__init__(in_features, out_features, mask, bias=True, device=None, dtype=None)[source]

Methods

`__init__`(in_features, out_features, mask[, ...])
`extra_repr`()
`forward`(input)
`reset_parameters`()