return inputShape;
}
- unsigned int newSizes[newNumDimensions];
+ std::vector<unsigned int> newSizes(newNumDimensions, 0);
unsigned int diff = newNumDimensions - inputShape.GetNumDimensions();
newSizes[i] = inputShape[i - diff];
}
- return TensorShape(newNumDimensions, newSizes);
+ return TensorShape(newNumDimensions, newSizes.data());
}
} // Anonymous namespace