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

偏执的太偏执、 2022-06-07 13:37 165阅读 0赞

在神经网络计算过程中，经常会遇到需要将矩阵中的某些元素取出并且单独进行计算的步骤（例如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

发表评论取消回复

表情：

评论列表（有 0 条评论，165人围观）

还没有评论，来说两句吧...

相关阅读

相关输入一个整数矩阵，计算位于矩阵边缘的元素之和。

![watermark_type_d3F5LXplbmhlaQ_shadow_50_text_Q1NETiBA6LaFMzM4_size_20_color_FFFFFF_t_7

刺骨的言语ヽ痛彻心扉/ 2023年09月29日 09:08/ 0 赞/ 21 阅读

相关将二位数组的相同下标的内容组成一个新的数组

将一个二位数组的相同下标的内容组成一个新的数组 //这个里面放二维数组 $arr = []; //定义一个空数组 $arr_new = arr

电玩女神/ 2023年07月13日 13:56/ 0 赞/ 5 阅读

相关如何从一个数组中提取某些属性或者做某些操作得到一个新的数组？

> 业精于勤荒于嬉 [Array.prototype.map()][Array.prototype.map] 数组的 map 方法创建一个新数组，其结果是该数组中的每

拼搏现实的明天。/ 2022年11月04日 11:24/ 0 赞/ 177 阅读

相关 matlab如何判断矩阵中元素都大于0,在MATLAB中找到矩阵中零元素的数量

![Image 1][] I have a NxM matrix for example named A. After some processes I want to co

￡神魔★判官ぃ/ 2022年08月30日 06:39/ 0 赞/ 337 阅读

相关输入一个整数矩阵，计算位于矩阵边缘的元素之和

输入一个整数矩阵，计算位于矩阵边缘的元素之和.所谓矩阵边缘的元素,就是第一行和最后一行的元素以及第一列和最后一列的元素。源代码： include<stdio.h>

妖狐艹你老母/ 2022年08月05日 13:14/ 0 赞/ 268 阅读

相关从JsonArray中取出最后三个对象组成新的JsonArray

JSONArray advice为： "[{name:'a',value:'aa'},{name:'b',value:'bb'},{name:'c',value:'c

小鱼儿/ 2022年06月11日 09:59/ 0 赞/ 196 阅读

相关如何将tensorflow.Variable中的某些元素取出组成一个新的矩阵

在神经网络计算过程中，经常会遇到需要将矩阵中的某些元素取出并且单独进行计算的步骤（例如MLE，Attention等操作）。那么在 tensorflow 的 Variable 类

偏执的太偏执、/ 2022年06月07日 13:37/ 0 赞/ 166 阅读

相关利用pytorch中的gather函数取出矩阵中的元素

最近想用pytorch搭一个机器翻译模型，过程中涉及到从矩阵中取出某些指定元素组成新矩阵的步骤，然后在pytorch官网上找到对应功能的函数为 torch.gather 。

心已赠人/ 2022年05月25日 20:48/ 0 赞/ 562 阅读

相关 matlab中如何删除向量和矩阵中的元素

分享一下我老师大神的人工智能教程！零基础，通俗易懂！[http://blog.csdn.net/jiangjunshow][http_blog.csdn.net_jiangju

男娘i/ 2022年04月18日 03:47/ 0 赞/ 1203 阅读

相关 js取出数组a里面的key 并组成新的数组

var val = [ { year: '2021', month: [] },

快来打我*/ 2021年07月25日 21:56/ 0 赞/ 506 阅读