Solved by anantguptadbl in post #2 
              
              
                @Takumi3 The monte carlo sounds interesting. Can you share a link of the paper/blog etc 
In your code, I can see an issue here 
In the predict function 
samples_ = []
for i in range(samples):
    pred = self.forward(x1, x2).detach()
    samples_ += pred

Although you are iterating over a range of 10…