batch_set_zero for np only

One should be able to call batch_set_zero for setting to zero only on np instead of np_part. This would be very convenient for tests with oct-test