Fix softmax bug

5b3d1cbe · hjchen2 · 5df51432 · 5b3d1cbe
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

src/operators/math/softmax.cpp src/operators/math/softmax.cpp +1 -1

未找到文件。
--- a/src/operators/math/softmax.cpp
+++ b/src/operators/math/softmax.cpp
@@ -128,7 +128,7 @@ void SoftmaxFuntor<CPU, float>::operator()(const framework::Tensor *X,
        x0 = vmulq_f32(x0, __inv_sum);
        x1 = vmulq_f32(x1, __inv_sum);
        vst1q_f32(output, x0);
-        vst1q_f32(output + 4, x0);
+        vst1q_f32(output + 4, x1);
      }
 #endif
      for (int i = 0; i < remain; ++i) {