Tensorflow進行多維矩陣的拆分與拼接實例
更新時間：2020年02月07日 10:02:38 作者：張叫張大衛(wèi)
今天小編就為大家分享一篇Tensorflow進行多維矩陣的拆分與拼接實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
最近在使用tensorflow進行網(wǎng)絡(luò)訓(xùn)練的時候，需要提取出別人訓(xùn)練好的卷積核的部分層的數(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)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
今天小編就為大家分享一篇Python 給某個文件名添加時間戳的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-10-10
pytorch實現(xiàn)特殊的Module--Sqeuential三種寫法
今天小編就為大家分享一篇pytorch實現(xiàn)特殊的Module--Sqeuential三種寫法。具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-01-01
這篇文章主要介紹了如何用Python寫一個簡單的通訊錄，對著幾串代碼感興趣的朋友一起來看看吧2021-08-08
這篇文章主要為大家詳細(xì)介紹了python中web框架的自定義創(chuàng)建，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-09-09
這篇文章主要介紹了使用 Python 實現(xiàn)文件遞歸遍歷的三種方式,非常不錯，具有一定的參考借鑒價值，需要的朋友參考下吧2018-07-07
這篇文章主要給大家介紹了關(guān)于Python編碼解碼之encode()函數(shù)的相關(guān)資料,Python的encode函數(shù)用于將字符串按照指定的編碼方式進行編碼,返回一個bytes類型的對象,需要的朋友可以參考下2023-07-07
圖像的輪廓檢測不論是機器視覺還是其他方面都有較大作用,下面這篇文章主要給大家介紹了關(guān)于如何使用Python?OpenCV提取物體輪廓的相關(guān)資料,文中通過實例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-05-05
在日常工作中,我們經(jīng)常需要處理大量的Excel文件,為了保護敏感數(shù)據(jù)的安全性,我們可能需要對這些文件進行加密,本文主要介紹了如何使用Python實現(xiàn)批量加密Excel文件,需要的可以參考下2023-11-11
這篇文章主要介紹了Python繪制的二項分布概率圖,涉及Python基于numpy、math的數(shù)值運算及matplotlib圖形繪制相關(guān)操作技巧,需要的朋友可以參考下2018-08-08
python中split(),?os.path.split()和os.path.splitext()的用法
本文主要介紹了python中split(),?os.path.split()和os.path.splitext()的用法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-02-02