为什么在使用tensorboard时需要add_graph()的第二个参数?

2024-09-27 14:24:26 发布

您现在位置:Python中文网/ 问答频道 /正文

我们知道,add_graph()有两个参数,第一个参数是我们想要可视化的模型对象,第二个参数只是模型的输入。输入可以是非常随意的,也就是说,我们可以简单地随机生成一个具有所需形状的输入,并将其提供给add_graph()。那么把输入作为参数有什么意义呢?在我看来,这个参数只提供了一个输入张量的形状信息,但是我们可以直接从模型定义中得到这个信息,对吗?不管我们给add_graph()输入什么,tensorboard打印的计算图总是一样的。我真的很困惑


Tags: 对象模型add信息参数定义可视化graph

热门问题