From abb28435d6fed9adc1e65ca7ddfbaf8f0b671af1 Mon Sep 17 00:00:00 2001 From: WuJinxuan <2456510228@qq.com> Date: Sat, 5 Nov 2022 15:38:15 +0800 Subject: [PATCH] fix:pnnx-softmax (#4333) --- tools/pnnx/src/pass_level2/F_softmax.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/pnnx/src/pass_level2/F_softmax.cpp b/tools/pnnx/src/pass_level2/F_softmax.cpp index af34a58957c..8a9352beba2 100644 --- a/tools/pnnx/src/pass_level2/F_softmax.cpp +++ b/tools/pnnx/src/pass_level2/F_softmax.cpp @@ -25,7 +25,7 @@ class F_softmax : public GraphRewriterPass 5 4 pnnx.Input input_0 0 1 input pnnx.Input input_1 0 1 dim -prim::Constant op_0 0 1 dtype value=None +prim::Constant op_0 0 1 dtype value=* aten::softmax op_1 3 1 input dim dtype out pnnx.Output output 1 0 out )PNNXIR";