今天小编就为大家分享一篇将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 中，如