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.h
+++ b/sherpa-onnx/csrc/hypothesis.h
@@ -121,6 +121,9 @@ class Hypotheses {
  Map hyps_dict_;
 };

+const std::vector<int32_t> GetHypsRowSplits(
+    const std::vector<Hypotheses> &hyps);
+
 }  // namespace sherpa_onnx

 #endif  // SHERPA_ONNX_CSRC_HYPOTHESIS_H_