How to add namespaces to a tensorboard graph?

When displaying a graph using torch.utils.tensorboard.SummaryWriter.add_graph, only modules end up in boxes. I use a lot of ops around it, and it can be really, really hard to tell what’s going on. Is there something like tensorflow1.0’s tf.namespace (might be variable_scope actually), which would let me sort operations to more clearly see what’s happening? Thanks.

@Nathan_Wood Have you solved this issue ?

I have similar issue at Adding node and cell names for tensorboard graph