mdlearn.nn.modules.conv2d_encoder

Classes

Conv2dEncoder(*args, **kwargs)

class mdlearn.nn.modules.conv2d_encoder.Conv2dEncoder(*args: Any, **kwargs: Any)

forward(x: torch.Tensor)

init_weights(init_weights: Optional[str])