-
Notifications
You must be signed in to change notification settings - Fork 285
Open
Description
This is neat and by solving an NP-hard problem in my head I can tell which node corresponds to a given other tensor in my code. Could you add the names of each tensor to its node in the graph? Perhaps using https://pypi.org/project/varname/ or by modifying Tensor.__init__
to extract variable names from declarations in the traceback, then saving them in grad_fn
.
Metadata
Metadata
Assignees
Labels
No labels