Browse Source

fix:pnnx-softmax (#4333)

tags/20221128
WuJinxuan GitHub 3 years ago
parent
commit
abb28435d6
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      tools/pnnx/src/pass_level2/F_softmax.cpp

+ 1
- 1
tools/pnnx/src/pass_level2/F_softmax.cpp View File

@@ -25,7 +25,7 @@ public:
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";


Loading…
Cancel
Save