When creating the labels vector, we incorrectly initialized it with the
number of tensors and used push_back afterwards, leading to twice as many
entries. Instead, we now reserve and use push_back.