Skip to content
Snippets Groups Projects
Commit 63779e04 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Merge branch 'fix_softmax_binding' into 'dev'

Fix softmax binding

See merge request !156
parents db2365b6 b2af0b56
No related branches found
No related tags found
2 merge requests!212Version 0.3.0,!156Fix softmax binding
Pipeline #50258 canceled
......@@ -21,7 +21,7 @@ namespace Aidge {
void init_Softmax(py::module& m) {
py::class_<Softmax_Op, std::shared_ptr<Softmax_Op>, OperatorTensor>(m, "SoftmaxOp", py::multiple_inheritance())
.def(py::init<std::size_t>(), py::arg("axis"))
.def(py::init<std::int32_t>(), py::arg("axis"))
.def_static("get_inputs_name", &Softmax_Op::getInputsName)
.def_static("get_outputs_name", &Softmax_Op::getOutputsName);
declare_registrable<Softmax_Op>(m, "SoftmaxOp");
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment