My data must be some integer data，how to do 

My data must be some integer data. I also want to use torchviz to draw pictures. So what do I need to modify in the source code? Or is this not feasible? If it is not feasible, please tell me a method that can be used in python. A library for drawing function call graphs and supporting integer data. Thank you.