Add an optional epsilon to avoid NANs when big and small are too close in computing log_prob.

Add an optional epsilon to avoid NANs when big and small are too close in computing log_prob. #1289