Log the sizes of inputs in the assert of setInputTensorTypes(...) in jit/python/script_init.cpp for easy debugging. Helps/close: https://github.com/pytorch/pytorch/issues/72763 Fixes #72763 Pull Request resolved: https://github.com/pytorch/pytorch/pull/80495 Approved by: https://github.com/davidberard98