Minimal setup for ONNX fail shape inference

import torch


class DummyModel(torch.nn.Module):
    def __init__(self):
        super().__init__()

    def forward(self, x):
        return torch.full([x], 0)


dummy_model = DummyModel()
dummy_data = torch.tensor(1, dtype=torch.int32)
torch.onnx.export(dummy_model, (dummy_data,), f="my_filename.onnx")