Tensorflow进行多维矩阵的拆分与拼接实例
更新时间：2020年02月07日 10:02:38 作者：张叫张大卫
今天小编就为大家分享一篇Tensorflow进行多维矩阵的拆分与拼接实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
最近在使用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进行多维矩阵的拆分与拼接实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
今天小编就为大家分享一篇Python 给某个文件名添加时间戳的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-10-10
pytorch实现特殊的Module--Sqeuential三种写法
今天小编就为大家分享一篇pytorch实现特殊的Module--Sqeuential三种写法。具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-01-01
这篇文章主要介绍了如何用Python写一个简单的通讯录，对着几串代码感兴趣的朋友一起来看看吧2021-08-08
这篇文章主要为大家详细介绍了python中web框架的自定义创建，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-09-09
这篇文章主要介绍了使用 Python 实现文件递归遍历的三种方式,非常不错，具有一定的参考借鉴价值，需要的朋友参考下吧2018-07-07
这篇文章主要给大家介绍了关于Python编码解码之encode()函数的相关资料,Python的encode函数用于将字符串按照指定的编码方式进行编码,返回一个bytes类型的对象,需要的朋友可以参考下2023-07-07
图像的轮廓检测不论是机器视觉还是其他方面都有较大作用,下面这篇文章主要给大家介绍了关于如何使用Python OpenCV提取物体轮廓的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下2022-05-05
在日常工作中,我们经常需要处理大量的Excel文件,为了保护敏感数据的安全性,我们可能需要对这些文件进行加密,本文主要介绍了如何使用Python实现批量加密Excel文件,需要的可以参考下2023-11-11
这篇文章主要介绍了Python绘制的二项分布概率图,涉及Python基于numpy、math的数值运算及matplotlib图形绘制相关操作技巧,需要的朋友可以参考下2018-08-08
python中split(), os.path.split()和os.path.splitext()的用法
本文主要介绍了python中split(), os.path.split()和os.path.splitext()的用法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-02-02
最新评论