Generate a Bernoulli dropout mask as a flat 1-D storage.

Each element is independently set to 1.0 with probability keep_prob and to 0.0 otherwise. The mask is generated on CPU using gen and then transferred to device via Dispatcher::from_cpu to keep the random number stream deterministic regardless of where the consuming op runs.