❓ [QUESTION] how to get the model.pth input shape ?

[WorldPeace123]

hello ,I want to us tvm to make the pth+lammps run better ,but i dont know how to get the model input?(i have the xyz,yaml allthat can run with lammps).if i can run the model with python ,i can make the model have better performance.so ,can you help me ?thanks !