import tensorflow as tf

  # rotates around z, while we rotate around y so need to swap
  center_1 = tf.reshape(box1[0:3][[0, 2, 1]], [1, 3])
  center_2 = tf.reshape(box2[0:3][[0, 2, 1]], [1, 3])

  rotation_z_1 = tf.reshape(box1[-1], [1])
  rotation_z_2 = tf.reshape(box2[-1], [1])

  length_1 = tf.reshape(box1[3 + 0], [1])
