tensorplay.utils.viz API

Render the computation graph recorded for var.

Leaf tensors with requires_grad=True are drawn light blue and are labeled by the name passed in params; the output tensor is light green; recorded operations are white and labeled by operation name.

Parameters:

• var – output tensor whose recorded chain should be drawn.

• params – dict mapping a name to a leaf tensor, used for labeling.

Returns a renderable object: a graphviz.Digraph when the graphviz package is installed, otherwise a matplotlib wrapper with the same render(filename, format="png") call from the networkx fallback.