| | |
|---|
| | | text_lengths_dim.size()); //, ONNX_TENSOR_ELEMENT_DATA_TYPE_INT32); |
|---|
| | | |
|---|
| | | //vad_mask |
|---|
| | | vector<float> arVadMask,arSubMask; |
|---|
| | | // vector<float> arVadMask,arSubMask; |
|---|
| | | vector<float> arVadMask; |
|---|
| | | int nTextLength = input_data.size(); |
|---|
| | | |
|---|
| | | VadMask(nTextLength, nCacheSize, arVadMask); |
|---|
| | | Triangle(nTextLength, arSubMask); |
|---|
| | | // Triangle(nTextLength, arSubMask); |
|---|
| | | std::array<int64_t, 4> VadMask_Dim{ 1,1, nTextLength ,nTextLength }; |
|---|
| | | Ort::Value onnx_vad_mask = Ort::Value::CreateTensor<float>( |
|---|
| | | m_memoryInfo, |
|---|
| | |
|---|
| | | std::array<int64_t, 4> SubMask_Dim{ 1,1, nTextLength ,nTextLength }; |
|---|
| | | Ort::Value onnx_sub_mask = Ort::Value::CreateTensor<float>( |
|---|
| | | m_memoryInfo, |
|---|
| | | arSubMask.data(), |
|---|
| | | arSubMask.size() , |
|---|
| | | arVadMask.data(), |
|---|
| | | arVadMask.size(), |
|---|
| | | SubMask_Dim.data(), |
|---|
| | | SubMask_Dim.size()); // , ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT); |
|---|
| | | |
|---|
