tf.compat.v1.assert_rank