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.