def METHOD_NAME(boxes):
    return tf.concat(
        [
            boxes["boxes"],
            tf.expand_dims(
                tf.cast(boxes["classes"], tf.float32), axis=-1
            ),
            tf.expand_dims(
                tf.cast(boxes["difficulty"], tf.float32), axis=-1
            ),
        ],
        axis=-1,
    )