标签: tensorflow
我想像keras.summary()一样在每层上打印tf模型的参数编号,因此我需要知道tf模型的层以及每层的参数信息。我知道tensorlayer具有count_params()函数,所以我只需要从一个存在的模型中获取每一层。怎么做?
keras.summary()
count_params()
ps,this code从模型中打印参数图,但是似乎不是我的 期望