Я экспортировал свою модель PyTorch в ONNX.Теперь, есть ли способ для меня получить входной слой от этой модели ONNX?
Экспорт модели PyTorch в ONNX
import torch.onnx
checkpoint = torch.load("./saved_pytorch_model.pth")
model.load_state_dict(checkpoint['state_dict'])
input = torch.tensor(df_X.values).float()
torch.onnx.export(model, input, "onnx_model.onnx")
Загрузка модели ONNX
onnx_model = onnx.load('onnx_model.onnx')
Я хочу иметь возможность каким-то образом получить входной слой из onnx_model.Возможно ли это?