operator issue in vector

This commit is contained in:
Shubham Agrawal
2022-06-10 14:37:06 +05:30
parent 6cf0a0eb7c
commit cdf2912074
2 changed files with 1 additions and 2 deletions
@@ -292,7 +292,6 @@ void LambdaMapReduceType<ReductionRuleType, MatType>::ComputeOutputDimensions()
this->outputDimensions = this->inputDimensions;
return;
}
network.front()->InputDimensions() = this->inputDimensions;
inSize = this->inputDimensions[0];
for (size_t i = 1; i < this->inputDimensions.size(); ++i)
inSize *= this->inputDimensions[i];
@@ -73,7 +73,7 @@ class AddReductionType
const std::vector<size_t> networkSize = network[0]->OutputDimensions();
for (size_t i = 1; i < network.size(); i++)
{
if (networkSize != network[i]->OutputDimensions())
if (!(networkSize == network[i]->OutputDimensions()))
{
Log::Fatal << "Network size mismatch." << std::endl;
}