improve bottleneck
improvement on generating dropout mask