def testRandomWalk(self): # Fully known shape. rnd1 = init_ops._random_walk([1, 2], tf.nn.relu) self.assertEqual([1, 2], rnd1.get_shape())