tensorplay.utils.viz

Functions

make_dot() [source]

make_dot(var, params=None)

Produces Graphviz representation of PyTorch graph. If a node is a Variable (requires_grad=True), it will be blue. If a node is an operation (grad_fn), it will be gray.

Args

• var: output Variable
• params: dict of (name, Variable) to add names to node that require grad (TODO: implement param naming)