tensorflow 獲取模型所有參數總和數量的方法
更新時間：2018年06月14日 09:19:07 作者：guotong1988
今天小編就為大家分享一篇tensorflow 獲取模型所有參數總和數量的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
實例如下所示：
from functools import reduce from operator import mul def 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 獲取模型所有參數總和數量的方法就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
有很多地方都說python多繼承的繼承順序,是按照深度遍歷的方式,其實python多繼承順序的算法,不是嚴格意義上的深度遍歷,而是基于深度遍歷基礎上優(yōu)化出一種叫3C算法,本文將給大家詳細的介紹一下python多繼承的3C算法,需要的朋友可以參考下2024-07-07
這篇文章主要給大家介紹了關于python中round()函數用法的相關資料,round()函數是Python內置函數之一,用于對數字進行四舍五入操作,需要的朋友可以參考下2023-08-08
這篇文章主要介紹了用?Python?繪制全國鴻星爾克門店分布圖，今天就以某度地圖?用Python爬蟲看一下全國到底有多少家鴻星爾克門店，,需要的朋友可以參考一下2022-01-01
Python中利用pyqt5制作指針鐘表顯示實時時間(指針時鐘)
這篇文章主要介紹了Python中利用pyqt5制作指針鐘表顯示實時時間(動態(tài)指針時鐘),本文通過示例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2022-02-02
最近做程序用到漢字轉漢語拼音的問題，首先想到Google一下，但是拿來看了一下源碼，不是甚清楚。2009-02-02
這篇文章主要介紹了python實現交并比IOU教程，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-04-04
python利用os模塊編寫文件復制功能——copy()函數用法
這篇文章主要介紹了python利用os模塊編寫文件復制功能——copy()函數用法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-07-07
這篇文章主要介紹了Python中的groupby分組功能的實例代碼，非常不錯，具有一定的參考借鑒價值，需要的朋友參考下吧2018-07-07
Python自定義聚合函數merge與transform區(qū)別詳解
這篇文章主要介紹了Python自定義聚合函數merge與transform區(qū)別詳解,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-05-05
這篇文章主要介紹了python繪制鉛球的運行軌跡代碼分享，具有一定參考價值，需要的朋友可以了解下。2017-11-11