Tensorflow進(jìn)行多維矩陣的拆分與拼接實(shí)例
最近在使用tensorflow進(jìn)行網(wǎng)絡(luò)訓(xùn)練的時(shí)候,需要提取出別人訓(xùn)練好的卷積核的部分層的數(shù)據(jù)。由于tensorflow中的tensor和python中的list不同,無法直接使用加法進(jìn)行拼接,后來發(fā)現(xiàn)一個(gè)函數(shù)可以完成tensor的拼接。
函數(shù)形式如下:
tf.concat(concat_dim,values,name='concat')
其中,第一個(gè)參數(shù)表示需要拼接的多維tensor,并且可以將多個(gè)tensor同事拼接,第二個(gè)表示按照哪一個(gè)維度拼接(從數(shù)字0開始)。
例子:創(chuàng)建一個(gè)三維的tensor,然后分別取出最后一個(gè)維度(注意:tensor支持與python中l(wèi)ist相似的切片操作,可以使用這種方式進(jìn)行拆分),然后在拼接在一起。
import tensorflow as tf weights=tf.Variable(tf.truncated_normal([2,3,4],dtype=tf.float32,stddev=1e-1),name='weights') weight1=weights[0:2,0:3,1:2] weight2=weights[0:2,0:3,2:3] weight3=weights[0:2,0:3,1:2] weight4=tf.concat([weight1,weight2,weight3],2) #2表示最后一個(gè)維度 with tf.Session() as sess: sess.run(tf.global_variables_initializer()) print(sess.run(weights)) print("****************") print(sess.run(weight4))
以上這篇Tensorflow進(jìn)行多維矩陣的拆分與拼接實(shí)例就是小編分享給大家的全部內(nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
Flask實(shí)現(xiàn)圖片的上傳、下載及展示示例代碼
這篇文章主要介紹了Flask實(shí)現(xiàn)圖片的上傳、下載及展示示例代碼,小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,也給大家做個(gè)參考。一起跟隨小編過來看看吧2018-08-08Python語法學(xué)習(xí)之進(jìn)程的創(chuàng)建與常用方法詳解
本文我們將學(xué)習(xí)一下在?Python?中去創(chuàng)建并使用多進(jìn)程的方法,可以通過創(chuàng)建多個(gè)進(jìn)程來幫助我們提高腳本執(zhí)行的效率,感興趣的可以了解一下2022-04-04利用Python實(shí)現(xiàn)微信找房機(jī)器人實(shí)例教程
這篇文章主要給大家介紹了關(guān)于如何利用Python實(shí)現(xiàn)微信找房機(jī)器人的相關(guān)資料,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者使用python具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面來一起學(xué)習(xí)學(xué)習(xí)吧2019-03-03在python 中split()使用多符號(hào)分割的例子
今天小編就為大家分享一篇在python 中split()使用多符號(hào)分割的例子,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-07-07Pandas的Series結(jié)構(gòu)及常用操作實(shí)例
這篇文章主要介紹了Pandas的Series結(jié)構(gòu)及常用操作實(shí)例,Series序列,是一種一維的結(jié)構(gòu),類似于一維列表和ndarray中的一維數(shù)組,但是功能比他們要更為強(qiáng)大,Series由兩部分組成:索引index和數(shù)值values,需要的朋友可以參考下2023-07-07ROS系統(tǒng)將python包編譯為可執(zhí)行文件的簡單步驟
本文章講述ROS系統(tǒng)下如何將python編譯為可以執(zhí)行文件,步驟比較簡單,感興趣的朋友跟隨小編一起看看吧2021-07-07python通過getopt模塊如何獲取執(zhí)行的命令參數(shù)詳解
這篇文章主要給大家介紹了關(guān)于python通過getopt模塊如何獲取執(zhí)行的命令參數(shù)的相關(guān)資料,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧。2017-12-12python利用beautifulSoup實(shí)現(xiàn)爬蟲
這篇文章主要介紹了python利用beautifulSoup實(shí)現(xiàn)爬蟲,需要的朋友可以參考下2014-09-09PyQt5下拉式復(fù)選框QComboCheckBox的實(shí)例
今天小編就為大家分享一篇PyQt5下拉式復(fù)選框QComboCheckBox的實(shí)例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-06-06