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实现的简单发送邮件脚本分享
Nov 07 Python
在Python的Flask框架中实现全文搜索功能
Apr 20 Python
Python批量修改文本文件内容的方法
Apr 29 Python
PyCharm在win10的64位系统安装实例
Nov 26 Python
我喜欢你 抖音表白程序python版
Apr 07 Python
33个Python爬虫项目实战(推荐)
Jul 08 Python
PyQt Qt Designer工具的布局管理详解
Aug 07 Python
pytorch中使用cuda扩展的实现示例
Feb 12 Python
Django使用Profile扩展User模块方式
May 14 Python
Python tkinter之ComboBox(下拉框)的使用简介
Feb 05 Python
python实现黄金分割法的示例代码
Apr 28 Python
如何通过一篇文章了解Python中的生成器
Apr 02 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
windows下PHP APACHE MYSQ完整配置
2007/01/02 PHP
PHP的变量总结 新手推荐
2011/04/18 PHP
php中++i 与 i++ 的区别
2012/08/08 PHP
基于PHP对XML的操作详解
2013/06/07 PHP
php中将一个对象保存到Session中的方法
2015/03/13 PHP
IE innerHTML,outerHTML所引起的问题
2009/06/04 Javascript
window.onbeforeunload方法在IE下无法正常工作的解决办法
2010/01/23 Javascript
JS的反射问题
2010/04/07 Javascript
JavaScript中的面向对象介绍
2012/06/30 Javascript
JS获取计算机mac地址以及IP的实现方法
2014/01/08 Javascript
css结合js制作下拉菜单示例代码
2014/02/27 Javascript
jQuery学习笔记之 Ajax操作篇(三) - 过程处理
2014/06/23 Javascript
jQuery中live()方法用法实例
2015/01/19 Javascript
浅谈jQuery的offset()方法及示例分享
2015/07/17 Javascript
基于Jquery实现万圣节快乐特效
2015/11/01 Javascript
js利用appendChild对标签进行排序的实现方法
2016/10/16 Javascript
JavaScript中object和Object的区别(详解)
2017/02/27 Javascript
微信小程序 使用腾讯地图SDK详解及实现步骤
2017/02/28 Javascript
vue.js实现条件渲染的实例代码
2017/06/22 Javascript
JS中offset和匀速动画详解
2018/02/06 Javascript
bootstrap模态框弹出和隐藏,动态改变中间内容的实例
2018/08/10 Javascript
使用jquery的cookie实现登录页记住用户名和密码的方法
2019/03/13 jQuery
javascript实现简易数码时钟
2020/03/30 Javascript
使用Webpack 搭建 Vue3 开发环境过程详解
2020/07/28 Javascript
Python性能优化技巧
2015/03/09 Python
使用Python实现微信提醒备忘录功能
2018/12/04 Python
Python一键安装全部依赖包的方法
2019/08/12 Python
PYTHON如何读取和写入EXCEL里面的数据
2019/10/28 Python
哈利波特商店:Harry Potter Shop
2018/11/30 全球购物
竞选班长演讲稿
2013/12/30 职场文书
七夕情人节促销方案
2014/06/07 职场文书
银行转正自我鉴定
2014/09/29 职场文书
情况说明书怎么写
2015/10/08 职场文书
MySQL系列之一 MariaDB-server安装
2021/07/02 MySQL
python3实现常见的排序算法(示例代码)
2021/07/04 Python
SQL中的三种去重方法小结
2021/11/01 SQL Server