diff --git a/audio/paddleaudio/src/pybind/kaldi/kaldi_feature_wrapper.cc b/audio/paddleaudio/src/pybind/kaldi/kaldi_feature_wrapper.cc index 3072c4070..8b8ff18be 100644 --- a/audio/paddleaudio/src/pybind/kaldi/kaldi_feature_wrapper.cc +++ b/audio/paddleaudio/src/pybind/kaldi/kaldi_feature_wrapper.cc @@ -30,7 +30,6 @@ bool KaldiFeatureWrapper::InitFbank(knf::FbankOptions opts) { py::array_t KaldiFeatureWrapper::ComputeFbank( const py::array_t wav) { py::buffer_info info = wav.request(); - //::kaldi::SubVector<::kaldi::BaseFloat> input_wav((float*)info.ptr, info.size); std::vector input_wav((float*)info.ptr, (float*)info.ptr + info.size); std::vector feats; @@ -39,10 +38,7 @@ py::array_t KaldiFeatureWrapper::ComputeFbank( auto result = py::array_t(feats.size()); py::buffer_info xs = result.request(); float* res_ptr = (float*)xs.ptr; - for (int idx = 0; idx < feats.size(); ++idx) { - *res_ptr = feats[idx]; - res_ptr++; - } + std::memcpy(res_ptr, feats.data(), sizeof(float)*feats.size()); std::vector shape{static_cast(feats.size() / Dim()), static_cast(Dim())}; return result.reshape(shape);