Tensorflow进行多维矩阵的拆分与拼接实例
作者:张叫张大卫 时间:2021-11-29 22:20:58
最近在使用tensorflow进行网络训练的时候,需要提取出别人训练好的卷积核的部分层的数据。由于tensorflow中的tensor和python中的list不同,无法直接使用加法进行拼接,后来发现一个函数可以完成tensor的拼接。
函数形式如下:
tf.concat(concat_dim,values,name='concat')
其中,第一个参数表示需要拼接的多维tensor,并且可以将多个tensor同事拼接,第二个表示按照哪一个维度拼接(从数字0开始)。
例子:创建一个三维的tensor,然后分别取出最后一个维度(注意:tensor支持与python中list相似的切片操作,可以使用这种方式进行拆分),然后在拼接在一起。
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))
来源:https://blog.csdn.net/weixin_40100431/article/details/82858085
标签:Tensorflow,多维,矩阵
0
投稿
猜你喜欢
使用Perl语言去存取mSQL和MySQL数据库的内容
2009-10-23 09:11:00
用Python写一个模拟qq聊天小程序的代码实例
2022-04-01 10:01:42
Python input函数使用实例解析
2022-01-05 21:00:34
几个javascript显示加载进度条代码
2008-01-19 10:52:00
Jenkins配置maven项目之打包、部署、发布的全过程
2023-08-07 19:14:29
深入了解Python装饰器的高级用法
2023-12-19 13:34:45
网页设计趋势之:”勾引”用户的按钮
2009-02-17 12:09:00
如何用ASP输出HTML文件?
2010-06-11 20:01:00
Python3.10新特性之match语句示例详解
2023-07-23 05:17:23
Python标准库使用OrderedDict类的实例讲解
2022-07-17 22:27:31
[译]在线广告及其在网页设计中的重要性
2009-02-06 13:36:00
Python字符串、元组、列表、字典互相转换的方法
2022-10-28 20:25:51
Python远程控制Windows服务器的方法详解
2023-07-13 13:14:03
asp如何从数据库中调出个人档案?
2009-11-15 20:02:00
Python 读写 Matlab Mat 格式数据的操作
2023-08-23 01:21:12
asp不用DSN也能连接EXCEL吗?
2009-10-29 12:25:00
巧妙的Sql函数日期处理方法
2009-05-25 17:59:00
Python中的TCP socket写法示例
2023-06-25 00:21:05
Python matplotlib绘制散点图配置(万能模板案例)
2023-08-02 23:04:18
php替换字符串中间字符为省略号的方法
2023-09-30 20:16:04