實例如下所示：

from functools import reducefrom operator import muldef get_num_params(): num_params = 0 for variable in tf.trainable_variables():  shape = variable.get_shape()  num_params += reduce(mul, [dim.value for dim in shape], 1) return num_params

以上這篇tensorflow 獲取模型所有參數總和數量的方法就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持武林站長站。