Add mini-batch feature.
Add mini-batch feature.