Я хочу, чтобы визуализатор графа пометил края тензорными размерами, а толщина края отражала бы общий тензорный размер. Практически точно так же, как написано в этом документе:
Когда сериализованный GraphDef включает тензорные формы, визуализатор графа помечает края тензорными размерами, а толщина края отражает общий размер тензора. Чтобы включить тензорные формы в GraphDef, передайте фактический объект графика (как в sessions.graph) в FileWriter при сериализации графика. На изображениях ниже показана модель CIFAR-10 с информацией о форме тензора:
Я передаю объект графика в свое резюме.
writer = tf.summary.FileWriter(_dir_tensorboard, graph=sess.graph, flush_secs=300)
Но я не получил никакой информации о толщине (все линии одного размера). У меня есть только информация о форме тензора и информация о количестве тензоров.
Как я могу добиться такого же визуального эффекта, как утверждается в руководстве?