不知何故,我最终得到的列表看起来像[ 1 36 2 72 37 74]
而不是[ 1, 36, 2, 72, 37,74]
。我如何转换它,以便我可以使用这些值来选择矩阵A的行,在我的情况下,这是一个5266 x 441矩阵?输出应为6 x 441矩阵。
答案 0 :(得分:1)
虽然我没有看到你的列表之间的区别(为什么一个有逗号,另一个没有?),我认为你可以使用tf.gather
函数来结束你想要的矩阵得到:https://www.tensorflow.org/api_docs/python/tf/gather