BayesConv2d

class torch_uncertainty.layers.bayesian.BayesConv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, prior_sigma_1=0.1, prior_sigma_2=0.002, prior_pi=1, mu_init=0.0, sigma_init=-6.0, frozen=False, groups=1, bias=True, padding_mode='zeros', device=None, dtype=None)

Bayesian Conv2d Layer with Gaussian Mixture prior and Normal posterior.