CNN 학습 가능 매개 변수의 수 - Python/TensorFlow

Question

TensorFlow에는 네트워크에서 학습 매개 변수의 양을 알아 내기 위해 할 수있는 기능이 있습니까?

Answer 1

나는 알고있다 기능 없음,하지만 여전히 자신이 tf.trainable_variables():

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)