将tensorflow.Variable中的某些元素取出组成一个新的矩阵示例
作者:耳东鹿其 发布时间:2022-01-17 23:49:50
在神经网络计算过程中,经常会遇到需要将矩阵中的某些元素取出并且单独进行计算的步骤(例如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
来源:https://blog.csdn.net/qq_32492561/article/details/78316742
猜你喜欢
- 我想从文件读数据的话,要比直接从数据库快一些吧(个人观点),昨天测试了读出6000条新闻,读数据库用了23579毫秒,读文件只用了123毫秒
- 本文实例讲述了Python类装饰器。分享给大家供大家参考,具体如下:编写类装饰器类装饰器类似于函数装饰器的概念,但它应用于类,它们可以用于管
- Rs.GetRows(N):N代表获取记录数量 Rs.GetRows(1):1表示只返回一行记录 Rs.GetRows(-1):-1表示默认
- *args和 **kwargs是常用的两个参数*args:用于接受多余的未命名的参数,元组类型。**kwargs:用于接受形参的命名参数,字
- 任何一个交互过程的操作,对于用户来说都有学习成本,谁也不能保证所有人都可以准确无误地走完一个流程。交互设计师在设计时应该考虑适时地给用户相应
- Variant变量一般会将其代表的数据子类型自动转换成合适的数据类型,但有时候,自动转换也会造成一些数据类型不匹配的错误.这时,可使用转换函
- 现在的离线浏览器很多,可是多数都是共享软件,或是英文软件,这给我们的操作带来了很大的不便。其实微软的网页制作工具FrontPage XP有一
- 本文实例讲述了Python实现矩阵加法和乘法的方法。分享给大家供大家参考,具体如下:本来以为python的矩阵用list表示出来应该很简单可
- 前言Pandas是为一次性处理整个行或列的矢量化操作而设计的,循环遍历每个单元格、行或列并不是它的设计用途。所以,在使用Pandas时,你应
- 一、前言Python语言近年来人气爆棚。它广泛应用于数据科学,人工智能,以及网络安全问题中,由于代码可读性较强,学习效率较高,吸引了许多非科
- 使用python爬虫其实就是方便,它会有各种工具类供你来使用,很方便。Java不可以吗?也可以,使用httpclient工具、还有一个大神写
- 拼音类文件py_class.php源码如下:<?php class py_class{ function py_class(){
- oracle服务器没有建立目标数据库的TNS时,在客户端(有权限的情况下)建立dblink语法如下: create database lin
- 前言博主参与过大大小小十次数学建模比赛,也获得了不少建模奖项。对于一些小批量样本数据去做预测或者是评估其规律性的话,比较适合的模型一般都是选
- 本文实例讲述了python实现中文转换url编码的方法。分享给大家供大家参考,具体如下:今天要处理百度贴吧的东西。想要做一个关键词的list
- 传参时传递可变对象,实际上传的是指向内存地址的指针/引用这个标题是我的结论,也是我在做项目过程查到的。学过C的都知道,函数传参可以传值,也可
- 方法对比:使用df[(df[“a”] > 3) & (df[“b&
- 关于跨域这个话题,很早就答应过要分享,但是因为懒,一直拖着,直到D2上有人谈起了“完美跨域”。“跨域”应该已经算不上什么难题了,只是提起“完
- 用python操作ms sqlserver,有好几种方法:(1)利用pymssql (2)利用pyodbc这里讲import&nb
- 对于Dreamweaver这个广大用户早有听闻的网页编辑工具,除功能强大外,Dreamweaver一向被推崇为同类网页编辑软件中产生垃圾代码