* refine doc of bernoulli

* fix some problems

* fix unsqueeze

* fix squeeze

* fix doc