x = np.random.normal(
        0, 1, np.prod(input_shape)).astype(np.float32).reshape(input_shape)