share GetHypsRowSplits interface and fix getting Topk not taking logprob (#131)

2023-04-26 11:41:04 +08:00
parent 86017f9833
commit aa7108729b
4 changed files with 52 additions and 38 deletions
--- a/sherpa-onnx/csrc/hypothesis.cc
+++ b/sherpa-onnx/csrc/hypothesis.cc
@@ -66,4 +66,19 @@ std::vector<Hypothesis> Hypotheses::GetTopK(int32_t k, bool length_norm) const {
  return {all_hyps.begin(), all_hyps.begin() + k};
 }

+const std::vector<int32_t> GetHypsRowSplits(
+    const std::vector<Hypotheses> &hyps) {
+  std::vector<int32_t> row_splits;
+  row_splits.reserve(hyps.size() + 1);
+
+  row_splits.push_back(0);
+  int32_t s = 0;
+  for (const auto &h : hyps) {
+    s += h.Size();
+    row_splits.push_back(s);
+  }
+
+  return row_splits;
+}
+
 }  // namespace sherpa_onnx