tensorflow 獲取模型所有參數(shù)總和數(shù)量的方法
更新時間：2018年06月14日 09:19:07 作者：guotong1988
今天小編就為大家分享一篇tensorflow 獲取模型所有參數(shù)總和數(shù)量的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
實例如下所示：
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 獲取模型所有參數(shù)總和數(shù)量的方法就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
我們刷視頻時常常會想把精彩的視頻保存到本地，如果少數(shù)的還行，如果有很多的話一個個保存太麻煩了。本文教你如何用Python實現(xiàn)視頻批量保存到本地，需要的可以參考一下2022-05-05
這篇文章主要介紹了Python中函數(shù)定義及其調(diào)用,感興趣的朋友可以來了解一下2021-06-06
Python帶動態(tài)參數(shù)功能的sqlite工具類
這篇文章主要介紹了Python帶動態(tài)參數(shù)功能的sqlite工具類,涉及Python針對sqlite數(shù)據(jù)庫的連接、查詢、sql語句執(zhí)行等相關(guān)操作封裝與使用技巧,需要的朋友可以參考下2018-05-05
這篇文章主要介紹了python中的編碼和解碼及\x和\u問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-05-05
Python 統(tǒng)計列表中重復(fù)元素的個數(shù)并返回其索引值的實現(xiàn)方法
這篇文章主要介紹了Python 統(tǒng)計列表中重復(fù)元素的個數(shù)并返回其索引值,本文給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-05-05
Python 實現(xiàn) T00ls 自動簽到腳本代碼（郵件+釘釘通知）
這篇文章主要介紹了Python 實現(xiàn) T00ls 自動簽到腳本（郵件+釘釘通知）,本文給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-07-07
這篇文章主要介紹了PyQt5/PySide2的按鈕控件使用實例，代碼簡單易懂，非常不錯，具有一定的參考借鑒價值,需要的朋友可以參考下2019-08-08
Django是基于MTV模式的框架，需要配合url控制器（路徑分發(fā)）使用，本文重點給大家介紹Django MTV模式詳解，需要的朋友參考下吧2021-09-09
這篇文章主要為大家介紹了python繼續(xù)找對象的方法，具有一定的參考價值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2022-01-01
如何將python的數(shù)據(jù)存儲到mysql數(shù)據(jù)庫中
在很多數(shù)據(jù)處理項目中,將數(shù)據(jù)存儲到數(shù)據(jù)庫中是非常常見的操作,下面這篇文章主要給大家介紹了關(guān)于如何將python的數(shù)據(jù)存儲到mysql數(shù)據(jù)庫中的相關(guān)資料,需要的朋友可以參考下2023-12-12