code: def _continue_test_batched(continue_trajectory, forward, reverse):
      with tf.name_scope("continue_test_batched"):
          return (
              continue_trajectory
              & _has_no_u_turn(
                  forward.state, reverse.state, forward.momentum
              )
              & _has_no_u_turn(
                  forward.state, reverse.state, reverse.momentum
              )
          )
  
imports: import collections
  import functools
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python import random
  from tensorflow_probability.python.experimental import auto_batching
  from tensorflow_probability.python.math import gradient
  from tensorflow_probability.python.mcmc import kernel
  from tensorflow_probability.python.util.seed_stream import SeedStream
label_desc: Use image node_id 12 for calling a global function with example usage: _continue_test_batched(continue_trajectory, forward, reverse) and returns: unknown