Create a wrapper for Conditional probability

[dhruvdcoder]

We need this because of the numerical issues with pytorch's softplus. While creating conditional probability we need to concatenate, take volume, and split again.

Optional step: Find the failure points: on what shapes and values does this happen?
First step: Profile the difference between applying softplus on two tensors or applying softplus on their concatenation and split it. Profile both runtime and memory usage.