From b11db185534471aca5a2ddb1bb97f512f301ef63 Mon Sep 17 00:00:00 2001 From: nihuini Date: Wed, 12 Feb 2025 16:33:30 +0800 Subject: [PATCH] w --- tools/pnnx/src/ir.cpp | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/tools/pnnx/src/ir.cpp b/tools/pnnx/src/ir.cpp index e79a09826d6..0fd83233175 100644 --- a/tools/pnnx/src/ir.cpp +++ b/tools/pnnx/src/ir.cpp @@ -2255,11 +2255,17 @@ int Graph::python(const std::string& pypath, const std::string& pnnxbinpath) { if (op->type == "Tensor.index_put" && it.first == "values") { - fprintf(pyfp, "torch.tensor(%g)", param.f); + if (param.f == (int)param.f) + fprintf(pyfp, "torch.tensor(%.1f)", param.f); + else + fprintf(pyfp, "torch.tensor(%g)", param.f); } else { - fprintf(pyfp, "%g", param.f); + if (param.f == (int)param.f) + fprintf(pyfp, "%.1f", param.f); + else + fprintf(pyfp, "%g", param.f); } } if (param.type == 4) @@ -2318,7 +2324,10 @@ int Graph::python(const std::string& pypath, const std::string& pnnxbinpath) fprintf(pyfp, "("); for (size_t i = 0; i < param.af.size(); i++) { - fprintf(pyfp, "%g", param.af[i]); + if (param.af[i] == (int)param.af[i]) + fprintf(pyfp, "%.1f", param.af[i]); + else + fprintf(pyfp, "%g", param.af[i]); if (i + 1 != param.af.size() || param.af.size() == 1) fprintf(pyfp, ","); }