I am seeking for example code of a simple Bayesian NN. Is it generally the similar way to implement with PyTorch as the traditional CNN does ?
yes, from what I understand about Bayesian NNs, you dont have to do anything drastically different in terms of implementation.