keras獲得model中某一層的某一個(gè)Tensor的輸出維度教程
獲得某層tensor的輸出維度
代碼如下所示:
from keras import backend as K @wraps(Conv2D) def my_conv(*args,**kwargs): new_kwargs={'kernel_regularizer':l2(5e-6)} new_kwargs['padding']='valid' #'same' new_kwargs['strides']=(2,2) if kwargs.get('strides')==(2,2) else (1,1) # new_kwargs['kernel_initializer']=keras.initializers.glorot_uniform(seed=0) new_kwargs.update(kwargs) return Conv2D(*args,**new_kwargs) def conv(x,**kwargs): x=my_conv(**kwargs)(x) x=BatchNormalization(axis=-1)(x) x=LeakyReLU(alpha=0.05)(x) return x def inception_resnet_a(x_input): x_short=x_input s1=conv(x_input,filters=32,kernel_size=(1,1)) s2=conv(x_input,filters=32,kernel_size=(1,1)) s2=conv(s2,filters=32,kernel_size=(3,3),padding='same') s3=conv(x_input,filters=32,kernel_size=(1,1)) s3=conv(s3,filters=48,kernel_size=(3,3),padding='same') s3=conv(s3,filters=64,kernel_size=(3,3),padding='same') x=keras.layers.concatenate([s1,s2,s3]) x=conv(x,filters=384,kernel_size=(1,1)) x=layers.Add()([x_short,x]) x=LeakyReLU(alpha=0.05)(x) print(K.int_shape(x))
使用K.int_shape(tensor_name)即可得到對(duì)應(yīng)tensor的維度
以上這篇keras獲得model中某一層的某一個(gè)Tensor的輸出維度教程就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
python程序文件擴(kuò)展名知識(shí)點(diǎn)詳解
在本篇文章里小編給大家整理的是關(guān)于python程序文件擴(kuò)展名知識(shí)點(diǎn)詳解內(nèi)容,需要的朋友們學(xué)習(xí)參考下。2020-02-02Pandas時(shí)間序列基礎(chǔ)詳解(轉(zhuǎn)換,索引,切片)
今天小編就為大家分享一篇Pandas時(shí)間序列基礎(chǔ)詳解(轉(zhuǎn)換,索引,切片),具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-02-02python實(shí)現(xiàn)學(xué)生信息管理系統(tǒng)(面向?qū)ο?
這篇文章主要介紹了python實(shí)現(xiàn)面向?qū)ο蟀鎸W(xué)生信息管理系統(tǒng),文中示例代碼介紹的非常詳細(xì),具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2022-06-06解決Python pandas df 寫(xiě)入excel 出現(xiàn)的問(wèn)題
今天小編就為大家分享一篇解決Python pandas df 寫(xiě)入excel 出現(xiàn)的問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-07-07Python機(jī)器學(xué)習(xí)工具scikit-learn的使用筆記
這篇文章主要介紹了Python機(jī)器學(xué)習(xí)工具scikit-learn的使用筆記,幫助大家更好的理解和使用python,感興趣的朋友可以了解下2021-01-01數(shù)組保存為txt, npy, csv 文件, 數(shù)組遍歷enumerate的方法
今天小編就為大家分享一篇數(shù)組保存為txt, npy, csv 文件, 數(shù)組遍歷enumerate的方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-07-07關(guān)于numpy中np.nonzero()函數(shù)用法的詳解
下面小編就為大家?guī)?lái)一篇關(guān)于numpy中np.nonzero()函數(shù)用法的詳解。小編覺(jué)得挺不錯(cuò)的,現(xiàn)在就分享給大家,也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2017-02-02