Doubly Stochastic Stream Mixing Forward Pass - Coding problem | OneNoughtOne