diff --git a/RecoTauTag/HLTProducers/src/L2TauTagNNProducerAlpaka.cc b/RecoTauTag/HLTProducers/src/L2TauTagNNProducerAlpaka.cc index 9772366c6b22e..9940650cfb7af 100644 --- a/RecoTauTag/HLTProducers/src/L2TauTagNNProducerAlpaka.cc +++ b/RecoTauTag/HLTProducers/src/L2TauTagNNProducerAlpaka.cc @@ -731,15 +731,19 @@ void L2TauNNProducerAlpaka::fillPatatracks(tensorflow::Tensor& cellGridMatrix, } std::vector L2TauNNProducerAlpaka::getTauScore(const tensorflow::Tensor& cellGridMatrix) { - std::vector pred_tensor; - tensorflow::run(L2cacheData_->session, {{inputTensorName_, cellGridMatrix}}, {outputTensorName_}, &pred_tensor); const int nTau = cellGridMatrix.shape().dim_size(0); - std::vector pred_vector(nTau); - for (int tau_idx = 0; tau_idx < nTau; ++tau_idx) { - pred_vector[tau_idx] = pred_tensor[0].matrix()(tau_idx, 0); - } + if (nTau == 0) { + return std::vector(); + } else { + std::vector pred_tensor; + tensorflow::run(L2cacheData_->session, {{inputTensorName_, cellGridMatrix}}, {outputTensorName_}, &pred_tensor); + std::vector pred_vector(nTau); + for (int tau_idx = 0; tau_idx < nTau; ++tau_idx) { + pred_vector[tau_idx] = pred_tensor[0].matrix()(tau_idx, 0); + } - return pred_vector; + return pred_vector; + } } void L2TauNNProducerAlpaka::produce(edm::Event& event, const edm::EventSetup& eventsetup) {