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