可训练变量获取

2017-08-08  本文已影响0人  Wangcy

tf.trainable_variables 获取所有可训练变量

tf.get_collection 用来获取制定集合(这里用来获取制定可训练变量),见图1

图 1
上一篇下一篇

猜你喜欢

热点阅读