Prevent half-precision mutual information overflow

Prevent half-precision mutual information overflow #860