将tensorflow.Variable中的某些元素取出组成一个新的矩阵示例
在神经网络计算过程中,经常会遇到需要将矩阵中的某些元素取出并且单独进行计算的步骤(例如MLE,Attention等操作)。那么在 tensorflow 的 Variable 类型中如何做到这一点呢?
首先假设 Variable 是一个一维数组 A:
import numpy as np import tensorflow as tf a = np.array([1, 2, 3, 4, 5, 6, 7, 8]) A = tf.Variable(a)
我们把我们想取出的元素的索引存到 B 中,如果我们只想取出数组 A 中的某一个元素,则 B 的设定为:
b = np.array([3]) B = tf.placeholder(dtype=tf.int32, shape=[1])
由于我们的索引坐标只有一维,所以 shape=1。
取出元素然后组合成tensor C 的操作如下:
C = tf.gather_nd(A, B)
运行:
init = tf.global_variables_initializer() with tf.Session() as sess: init.run() feed_dict = {B: b} result = sess.run([C], feed_dict=feed_dict) print result
得到:
[4]
如果我们想取出一维数组中的多个元素,则需要把每一个想取出的元素索引都单独放一行:
b = np.array([[3], [2], [5], [0]]) B = tf.placeholder(dtype=tf.int32, shape=[4, 1])
此时由于我们想要从一维数组中索引 4 个数,所以 shape=[4, 1]
再次运行得到:
[4 3 6 1]
////////////////////////////////////////////////////////////////////////////////////华丽丽的分割线
假设 Variable 是一个二维矩阵 A:
a = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]]) A = tf.Variable(a)
首先我们先取出 A 中的一个元素,需要给定该元素的行列坐标,存到 B 中:
b = np.array([2,3]) B = tf.placeholder(dtype=tf.int32, shape=[2])
注意由于我们输入的索引坐标变成了二维,所以shape也变为2。
取出元素然后组合成tensor C:
C = tf.gather_nd(A, B)
运行:
init = tf.global_variables_initializer() with tf.Session() as sess: init.run() feed_dict = {B: b} result = sess.run([C], feed_dict=feed_dict) print result
得到:
[12]
同样的,如果我们想取出二维矩阵中的多个元素,则需要把每一个想取出的元素的索引都单独放一行:
b = np.array([[2, 3], [1, 0], [2, 2], [0, 1]]) B = tf.placeholder(dtype=tf.int32, shape=[4, 2])
此时由于我们想要从二维矩阵中索引出 4 个数,所以 shape=[4, 2]
再次运行得到:
[12 5 11 2]
////////////////////////////////////////////////////////////////////////////////////华丽丽的分割线
推广到 n 维矩阵中:
假设 A 是 Variable 类型的 n 维矩阵,我们想取出矩阵中的 m 个元素,那么首先每个元素的索引坐标要表示成列表的形式:
index = [x1, x2, x3, ..., xn]
其中 xj 代表该元素在 n 维矩阵中第 j 维的位置。
其次每个坐标要单独占索引矩阵的一行:
index_matrix = [[x11, x12, x13, ..., x1n], [x21, x22, x23, ..., x2n], [x31, x32, x33, ..., x3n], ......................................., [xm1, xm2, xm3, ..., xmn]]
最后用 tf.gather_nd() 函数替换即可:
result = tf.gather_nd(A, index_matrix)
////////////////////////////////////////////////////////////////////////////////////华丽丽的分割线
[注] 问题出自:https://stackoverflow.com/questions/44793286/slicing-tensorflow-tensor-with-tensor
以上这篇将tensorflow.Variable中的某些元素取出组成一个新的矩阵示例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持猪先飞。
相关文章
- 这篇文章主要介绍了Python导入数值型Excel数据并生成矩阵操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-06-09
- 这篇文章主要为大家详细介绍了C++ Eigen库计算矩阵特征值及特征向量,具有一定的参考价值,感兴趣的小伙伴们可以参考一下...2020-04-25
- 这篇文章主要介绍了R语言创建矩阵的实现方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-05-06
- 今天小编就为大家分享一篇解决tensorflow训练时内存持续增加并占满的问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-04-22
从Pytorch模型pth文件中读取参数成numpy矩阵的操作
这篇文章主要介绍了从Pytorch模型pth文件中读取参数成numpy矩阵的操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-03-04- 这篇文章主要给大家介绍了关于Java实现矩阵乘法以及优化的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-02-05
win10安装tensorflow-gpu1.8.0详细完整步骤
这篇文章主要介绍了win10安装tensorflow-gpu1.8.0详细完整步骤,本文给大家介绍的非常详细,具有一定的参考借鉴价值,需要的朋友可以参考下...2020-04-22macOS M1(AppleSilicon) 安装TensorFlow环境
苹果为M1芯片的Mac提供了TensorFlow的支持,本文主要介绍了如何给使用M1芯片的macOS安装TensorFlow的环境,感兴趣的可以了解一下...2021-08-13- 这篇文章主要介绍了R语言 出现矩阵/缺失值的解决方案,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-05-06
windows系统Tensorflow2.x简单安装记录(图文)
这篇文章主要介绍了windows系统Tensorflow2.x简单安装记录(图文),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-01-18- 这篇文章主要介绍了C#计算矩阵的逆矩阵方法,较为详细的分析了逆矩阵的计算原理与相关的C#实现技巧,具有一定参考借鉴价值,需要的朋友可以参考下...2020-06-25
完美解决TensorFlow和Keras大数据量内存溢出的问题
这篇文章主要介绍了完美解决TensorFlow和Keras大数据量内存溢出的问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-07-04- 今天小编就为大家分享一篇Tensorflow读取并输出已保存模型的权重数值方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看不看...2020-04-30
在tensorflow下利用plt画论文中loss,acc等曲线图实例
这篇文章主要介绍了在tensorflow下利用plt画论文中loss,acc等曲线图实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-06-16Python通过TensorFLow进行线性模型训练原理与实现方法详解
这篇文章主要介绍了Python通过TensorFLow进行线性模型训练原理与实现方法,结合实例形式详细分析了Python通过TensorFLow进行线性模型训练相关概念、算法设计与训练操作技巧,需要的朋友可以参考下...2020-04-27详解tf.device()指定tensorflow运行的GPU或CPU设备实现
这篇文章主要介绍了详解tf.device()指定tensorflow运行的GPU或CPU设备实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-02-20- 今天小编就为大家分享一篇tensorflow实现对张量数据的切片操作方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-04-22
- 这篇文章主要介绍了C#实现矩阵加法、取负、数乘、乘法的方法,涉及C#针对矩阵的数学运算相关实现技巧,需要的朋友可以参考下...2020-06-25
- 这篇文章主要介绍了基于tensorflow for循环 while循环案例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-06-30
- 利用C++实现矩阵的相加/相称/转置/求鞍点。需要的朋友可以过来参考下,希望对大家有所帮助...2020-04-25