Mix distribution partially works

Test on cifar with batch-image distribution
Fill grad with 0 to make write functions well