python - CNN 可学习参数的数量 - Python/TensorFlow

Question

在 TensorFlow 中，是否有任何功能可以让我找出网络中学习参数的数量？

Answer 1

没有我知道的功能，但您仍然可以在 tf.trainable_variables():

上使用 for 循环计算自己

total_parameters = 0
for variable in tf.trainable_variables():
    variable_parameters = 1
    for dim in variable.get_shape():
        variable_parameters *= dim.value
    total_parameters += variable_parameters

print("Total number of trainable parameters: %d" % total_parameters)