a_np = (
        np.random.uniform(low=-1.0, high=1.0, size=np.prod(shape))