Tensorflow进行多维矩阵的拆分与拼接实例


Posted in Python onFebruary 07, 2020

最近在使用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))

Tensorflow进行多维矩阵的拆分与拼接实例

以上这篇Tensorflow进行多维矩阵的拆分与拼接实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
常见的python正则用法实例讲解
Jun 21 Python
python 专题九 Mysql数据库编程基础知识
Mar 16 Python
Python pymongo模块常用操作分析
Sep 01 Python
在PyCharm导航区中打开多个Project的关闭方法
Jan 17 Python
python实现贪吃蛇小游戏
Mar 21 Python
在python 中split()使用多符号分割的例子
Jul 15 Python
Django models.py应用实现过程详解
Jul 29 Python
pandas和spark dataframe互相转换实例详解
Feb 18 Python
pytorch masked_fill报错的解决
Feb 18 Python
django之导入并执行自定义的函数模块图解
Apr 01 Python
拿来就用!Python批量合并PDF的示例代码
Aug 10 Python
利用Python实现翻译HTML中的文本字符串
Jun 21 Python
Tensorflow训练模型越来越慢的2种解决方案
Feb 07 #Python
详解python itertools功能
Feb 07 #Python
Python中itertools的用法详解
Feb 07 #Python
Python转换itertools.chain对象为数组的方法
Feb 07 #Python
已安装tensorflow-gpu,但keras无法使用GPU加速的解决
Feb 07 #Python
python十进制转二进制的详解
Feb 07 #Python
基于Tensorflow使用CPU而不用GPU问题的解决
Feb 07 #Python
You might like
php&java(一)
2006/10/09 PHP
php获取mysql版本的几种方法小结
2008/03/25 PHP
php地址引用(php地址引用的效率问题)
2012/03/23 PHP
php实现的一个简单json rpc框架实例
2015/03/30 PHP
PHP实现远程下载文件到本地
2015/05/17 PHP
php简单实现发送带附件的邮件
2015/06/10 PHP
PHP实现事件机制实例分析
2015/06/26 PHP
文件上传之SWFUpload插件(代码)
2015/07/30 PHP
php安装ssh2扩展的方法【Linux平台】
2016/07/20 PHP
Laravel监听数据库访问,打印SQL的例子
2019/10/24 PHP
javascript 类型判断代码分析
2010/03/28 Javascript
javaScript年份下拉列表框内容为当前年份及前后50年
2014/05/28 Javascript
分享12个实用的jQuery代码片段
2016/03/09 Javascript
JS随机洗牌算法之数组随机排序
2016/03/23 Javascript
Vuejs第七篇之Vuejs过渡动画案例全面解析
2016/09/05 Javascript
打字效果动画的4种实现方法(超简单)
2017/10/18 Javascript
JS伪继承prototype实现方法示例
2018/06/20 Javascript
详解webpack自定义loader初探
2018/08/29 Javascript
Typescript3.9 常用新特性一览(推荐)
2020/05/14 Javascript
详解Vue Cli浏览器兼容性实践
2020/06/08 Javascript
js实现点击选项置顶动画效果
2020/08/25 Javascript
在vue中嵌入外部网站的实现
2020/11/13 Javascript
Python set集合类型操作总结
2014/11/07 Python
利用Python中unittest实现简单的单元测试实例详解
2017/01/09 Python
Python编程实现的简单神经网络算法示例
2018/01/26 Python
用sqlalchemy构建Django连接池的实例
2019/08/29 Python
Django实现文件上传下载
2019/10/06 Python
基于Keras 循环训练模型跑数据时内存泄漏的解决方式
2020/06/11 Python
HTML5使用DOM进行自定义控制示例代码
2013/06/08 HTML / CSS
预订奥兰多和佛罗里达州公园门票:FloridaTix
2018/01/03 全球购物
英国户外玩具儿童游乐设备网站:TP Toys(蹦床、攀爬框架、秋千、滑梯和游戏屋)
2018/04/09 全球购物
美国农场鲜花速递:The Bouqs
2018/07/13 全球购物
会计实习生自我鉴定
2013/12/12 职场文书
寒假思想汇报
2014/01/10 职场文书
农村婚礼司仪主持词
2015/06/29 职场文书
2016党员党课心得体会
2016/01/07 职场文书