Okay, thanks for the clarification.
I have some grad_fns like ‘Slice backward’ in my pytorchviz graph, which I am not sure how it works. Where can I find what the function does or I need to get what it does on the basis of the operation in the forward pass?
Thanks