How to get detail info of aten ops in a network

Hi, I want to grab aten ops information in a model, such as input shapes and other arguments.
I know two tools similarly:

  1. torch.profiler can log op input shapes, but do not include other arguments.
  2. torch-summary can log module info, but do not include ops info.

So how can I get these information? Thanks a lot!