Update probabilistic_max_pooling.py

added rng functions