Replacing nested loop with tensorflow.map_fn: “TypeError: List of Tensors when single Tensor expected”

Question

I have the following code, where I have two arrays, the first one is (bag_predictions) that contains the predictions of each bag, each bag contains several instances so the