Как правило, вы можете часто использовать скаляр вместо вектора, и это будет интерпретироваться как вектор, заполненный этим значением (например, 1 интерпретируется как 1,1,1,1).
Таким образом, если результат вашей функции потерь для партии равен x, он интерпретируется так, как если бы вы говорили, что потеря для каждого элемента в пакете равна x.