Skip to content
Snippets Groups Projects
Commit b055ebf0 authored by Maxence Naud's avatar Maxence Naud
Browse files

[Fix] input unsqueeze during output dimensions computation process in MatMul_Op

parent ecbcec98
No related branches found
No related tags found
No related merge requests found
......@@ -47,10 +47,10 @@ void Aidge::MatMul_Op::computeOutputDims() {
if (dims0.size() > dims1.size()) {
dims1.insert(dims1.cbegin(), dims0.begin(), dims0.end() - dims1.size());
dims1.insert(dims1.cbegin(), dims0.size() - dims1.size(), std::size_t(1));
}
else if (dims1.size() > dims0.size()) {
dims0.insert(dims0.cbegin(), dims1.begin(), dims1.end() - dims0.size());
dims0.insert(dims0.cbegin(), dims1.size() - dims0.size(), std::size_t(1));
}
AIDGE_ASSERT(dims0[dims_size-1] == dims1[dims_size-2], "Incompatible matrices sizes.");
......
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