将tensorflow.Variable中的某些元素取出组成一个新的矩阵示例


Posted in Python onJanuary 04, 2020

在神经网络计算过程中,经常会遇到需要将矩阵中的某些元素取出并且单独进行计算的步骤(例如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 相关文章推荐
python设置windows桌面壁纸的实现代码
Jan 28 Python
用实例分析Python中method的参数传递过程
Apr 02 Python
pygame学习笔记(3):运动速率、时间、事件、文字
Apr 15 Python
python生成IP段的方法
Jul 07 Python
python简单图片操作:打开\显示\保存图像方法介绍
Nov 23 Python
python 遍历目录(包括子目录)下所有文件的实例
Jul 11 Python
浅谈Python里面小数点精度的控制
Jul 16 Python
Python多进程池 multiprocessing Pool用法示例
Sep 07 Python
pandas的to_datetime时间转换使用及学习心得
Aug 11 Python
Python turtle库绘制菱形的3种方式小结
Nov 23 Python
pandas使用函数批量处理数据(map、apply、applymap)
Nov 27 Python
python常见的占位符总结及用法
Jul 02 Python
tensorflow实现tensor中满足某一条件的数值取出组成新的tensor
Jan 04 #Python
对tensorflow中的strides参数使用详解
Jan 04 #Python
tensorflow之获取tensor的shape作为max_pool的ksize实例
Jan 04 #Python
TensorFlow tf.nn.max_pool实现池化操作方式
Jan 04 #Python
TensorFlow tf.nn.conv2d实现卷积的方式
Jan 03 #Python
Python调用钉钉自定义机器人的实现
Jan 03 #Python
pytorch中的上采样以及各种反操作,求逆操作详解
Jan 03 #Python
You might like
PHP版网站缓存加快打开速度的方法分享
2012/06/03 PHP
PHP的pcntl多进程用法实例
2015/03/19 PHP
JavaScript中的闭包原理分析
2010/03/08 Javascript
js获得鼠标的坐标值的方法
2013/03/13 Javascript
详解Javascript 装载和执行
2014/11/17 Javascript
js淡入淡出焦点图幻灯片效果代码分享
2015/09/08 Javascript
Javascript点击其他任意地方隐藏关闭DIV实例
2016/06/21 Javascript
简单理解vue中track-by属性
2016/10/26 Javascript
关于JS Lodop打印插件打印Bootstrap样式错乱问题的解决方案
2016/12/23 Javascript
JS原生带小白点轮播图实例讲解
2017/07/22 Javascript
详解基于webpack2.x的vue2.x的多页面站点
2017/08/21 Javascript
vue router学习之动态路由和嵌套路由详解
2017/09/21 Javascript
js 开发之autocomplete="off"在chrom中失效的解决办法
2017/09/28 Javascript
jQuery中each方法的使用详解
2018/03/18 jQuery
jQuery替换节点元素的操作方法
2018/03/18 jQuery
vue.js前后端数据交互之提交数据操作详解
2018/04/24 Javascript
vue-router命名视图的使用讲解
2019/01/19 Javascript
了解JavaScript函数中的默认参数
2019/05/30 Javascript
初试vue-cli使用HBuilderx打包app的坑
2019/07/17 Javascript
Python字符串特性及常用字符串方法的简单笔记
2016/01/04 Python
pyqt 实现QlineEdit 输入密码显示成圆点的方法
2019/06/24 Python
python SVM 线性分类模型的实现
2019/07/19 Python
Python 多线程,threading模块,创建子线程的两种方式示例
2019/09/29 Python
Python PyQt5整理介绍
2020/04/01 Python
Vision Directa智利眼镜网:框架眼镜、隐形眼镜和名牌太阳眼镜
2016/11/23 全球购物
一套.net面试题及答案
2016/11/02 面试题
奥巴马演讲稿
2014/01/08 职场文书
学生干部学习的自我评价
2014/02/18 职场文书
2014年前台接待工作总结
2014/12/05 职场文书
2014个人年终工作总结范文
2014/12/15 职场文书
助学感谢信范文
2015/01/21 职场文书
挂职个人工作总结
2015/03/05 职场文书
中学校园广播稿
2015/08/18 职场文书
就业指导讲座心得体会
2016/01/15 职场文书
分析并发编程之LongAdder原理
2021/06/29 Java/Android
python分分钟绘制精美地图海报
2022/02/15 Python