The original model didn't batch features, instead, it iterates the list to predict a single input at once.
Upon stacking the input features, we can make the model more efficient.
Co-authored-by: Nwuzewu <wuzewu@baidu.com>
Co-authored-by: NLinjie Chen <40840292+linjieccc@users.noreply.github.com>