python取出矩阵中的某一元素_将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取出矩阵中的某一元素_将tensorflow.Variable中的某些元素取出组成一个新的矩阵示例...相关推荐
- python取矩阵的一部分_Python 实现取矩阵的部分列,保存为一个新的矩阵方法
Python 实现取矩阵的部分列,保存为一个新的矩阵方法 首先输入一个矩阵: >>> b=[[1,2,3,4,5,6],[2,2,3,4,5,6],[3,2,3,4,5,6],[4, ...
- python矩阵中找满足条件的元素_从numpy数组中取出满足条件的元素示例
例如问题:从 arr 数组中提取所有奇数元素. input:arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) output: #> array([1, ...
- lrange是取出所有值并移除么_图解双指针 | LeetCode 27. 移除元素
题目描述 原题链接:LeetCode 27. 移除元素 给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度. 不要使用额外的数组空间,你必须在 ...
- java 查找链表中间元素_如何在Java中一次性查找Java中链表的中间元素
如何在一次传递中找到LinkedList的中间元素?这是一个 Java 和非Java程序员面试时经常被问到的编程问题.这个问题类似于检查回文或计算阶乘,有时也会要求编写代码.为了回答这个问题,候选人必 ...
- python怎么从数组中提取连续的数字_从numpy数组中删除连续的数字
我是python新手,但我对我要做的这个项目真的很困惑.我看到了一张图片here.我要做的是找出图像中像素范围从0到255的所有方块的平均值.下面的代码显示了我用来计算图片值的方法.出现的问题是像素/ ...
- java数组删除数组元素_如何在Java中删除数组元素
java数组删除数组元素 When we create an array in Java, we specify its data type and size. This is used by JVM ...
- jquery 查找祖先元素_如何获得jQuery中元素的第一个祖先
jquery 查找祖先元素 In this post, we will discuss about the traversal method to get the first ancestor of ...
- jquery 获取同级元素_如何在jQuery中获取元素的同级
jquery 获取同级元素 In this post, we will discuss how to get the siblings of an HTML element in jQuery. jQ ...
- python读取视频里面爱豆出现的时间_计算任意视频中各人物的出镜时间(附Python实现)...
编译:Bing编者按:本文作者Pulkit Sharma分享了一篇有趣的项目,以<猫和老鼠>为例,计算任意视频中汤姆猫和杰瑞鼠的出镜时长.这一模型也可以用于其他电影,轻松统计各演员的上镜时 ...
最新文章
- osi七层网络层_OSI层速成课程
- Python 图像处理 | 图像平滑之均值滤波、方框滤波、高斯滤波及中值滤波
- 麻省理工学院的牛人解说数学体系,你到哪个层次了?
- list排序方法python_python list 排序的两种方法及实例讲解
- 配置nginx支持php
- YumRepo Error: All mirror URLs are not using ftp, http[s] or file. Eg. /HTML/ 标签: yum 2014-05-26
- 【剑指offer】_16 构建乘积数组
- Python网页抓取、模拟登录
- linux 下的sar工具命令小结
- phonegap走起
- SVN提示is already locked 解决办法
- SQL Server 历史SQL执行记录
- php私章制作,PS制作一枚私人的古典型印章教程
- 2022年11月广东软考系统集成真题及答案解析
- 布谷鸟优化算法 matlab,布谷鸟算法(Cuckoo Search,CS)MATLAB案例详细解析
- 计算机科学发展与大规模集成电路,计算机科学与技术发展趋势
- blog增量追加ing~(1/10阶段)ctf oj
- TCP三次握手第三次握手时ACK丢失怎么办
- 笔记本计算机声音小,笔记本没有声音,小编教你笔记本没有声音怎么修复
- 十款常用Linux系统介绍
热门文章
- Java线程的5种状态及切换
- go语言调度器源代码情景分析之五:汇编指令
- 做个程序员身体调查,希望大家点一下!
- 2021甘肃平凉高考成绩查询,甘肃省教育考试院:2021年甘肃高考查分入口、查分系统...
- 怎么判断我选了多少个复选框_7~8个月宝宝一天吃多少辅食,怎么安排?妈妈这样做,养出健康娃...
- golang mysql 回调_Golang操作MySQL的正确姿势
- pacman吃豆人_通过Tensorflow和强化学习实现自动化吃豆人PacMan
- pythonos模块介绍_python的os模块fnmatch模块介绍
- prometheus命令_Prometheus+Grafana 基础及简单搭建
- java年轻代频繁gc_年轻代频繁ParNew GC,导致http服务rt飙高