torch.contrib._graph_vis Namespace Reference

Functions
def	write (self, filename)

Detailed Description

Experimental. Tools for visualizing the torch.jit.Graph objects.

Function Documentation

def torch.contrib._graph_vis.write	(		self,
			filename
	)

Write an html file that visualizes a torch.jit.Graph using vis.js
Arguments:
    self (torch.jit.Graph): the graph.
    filename (string): the output filename, an html-file.

Definition at line 51 of file _graph_vis.py.