Update conv2d.py

added rng functions