Передать индексы единиц в цель в функцию потери для прогнозирования нескольких меток в Pytorch

Question

Рассмотрим следующую ситуацию для классификации по нескольким меткам.

output = torch.tensor([[0.3416, 0.6336, 0.3775, 0.2556, 0.6288]])
target = torch.tensor([[1,0,1,0,1]])
target_indices = torch.tensor([[0,2,4]])

Есть ли в Pytorch функция потерь, в которой я могу рассчитать CrossEntropyLoss следующим образом: loss(output, target_indices)?