We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fd48a6d commit 5e39de9Copy full SHA for 5e39de9
tmva/sofie/inc/TMVA/ROperator_Tanh.hxx
@@ -57,8 +57,10 @@ public:
57
std::stringstream out;
58
size_t length = ConvertShapeToLength(fShape);
59
out << "\n//------ TANH\n";
60
- out << SP << "for (int id = 0; id < " << length << " ; id++){\n";
61
- out << SP << SP << "tensor_" << fNY << "[id] = std::tanh(tensor_" << fNX << "[id]);\n";
+ out << SP << "for (size_t id = 0; id < " << length << " ; id++){\n";
+ out << SP << SP << "auto x = tensor_" << fNX << "[id];\n";
62
+ out << SP << SP << "auto exp2x = std::exp(2.0f * x);\n";
63
+ out << SP << SP << "tensor_" << fNY << "[id] = (exp2x - 1.0f) / (exp2x + 1.0f);\n";
64
out << SP << "}\n";
65
return out.str();
66
}
0 commit comments