MatShape realOutputDims = shape(result);

size_t realProd = std::accumulate(realOutputDims.begin(), realOutputDims.end(), 1, std::multiplies<int>());

// reduce dimentions

result = result.reshape(1, einsumOutDims.size(), einsumOutDims.data());

// Covered by ExplicitEinsumAsTensorContractionReshapeFinal.

output = output.reshape(1, reshaped_dims.size(), reshaped_dims.data());

}

}

} else { // This is the final pair - Transpose directly to the output ordering required and copy the contents to the op's output

// not sure if this finalize shape is needed at all