code: def test_double_backprop_make_dot(self):
      model, x = make_mlp_and_input()
      x.requires_grad = True
  
      def double_backprop(inputs, net):
          y = net(x).mean()
          (grad,) = torch.autograd.grad(
              y, x, create_graph=True, retain_graph=True
          )
          return grad.pow(2).mean() + y
  
      dot = make_dot(
          double_backprop(x, model),
          params=dict(list(model.named_parameters()) + [("x", x)]),
      )
  
file_path: pytorchviz/test/test.py
imports: import unittest
  import torch
  from torch import nn
  from torchviz import make_dot, make_dot_from_trace
label_desc: Use image node_id 2 for calling the TestTorchviz obj's underlying member method code with example usage: obj.test_double_backprop_make_dot() and returns: unknown