tf.debugging.assert_rank_in