将批量维度添加到 ONNX 模型答案

【问题标题】：Add Batch Dimension to ONNX model将批量维度添加到 ONNX 模型
【发布时间】：2021-11-25 05:24:54
【问题描述】：

我有 2 个 onnx 模型。第一个模型是较早训练的，我无法访问已保存模型的 pytorch 版本。模型输入的形状在图像中：Model 1。该模型对于模型的形状只有 1 个参数，并且没有批量大小的空间。我希望模型理想地具有像this 这样的输入。该模型具有 7 个变量的相同输入，但也有一个批量大小参数。我曾尝试使用 onnx 的更新模型工具，但它没有为批量大小添加额外的维度。

【问题讨论】：

标签： pytorch onnx

【解决方案1】：

据我所知，任何工具都不支持向现有 ONNX 模型添加批处理维度。实际上，复杂模型很难实现，因为它需要知道何时/如何为每个节点添加批处理维度。更好的方法仍然是在转换为 ONNX 之前添加批处理。

【讨论】：