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