使用Tensorflow

时间:2017-05-01 06:52:42

标签: tensorflow

不知何故,我最终得到的列表看起来像[ 1 36 2 72 37 74]而不是[ 1, 36, 2, 72, 37,74]。我如何转换它,以便我可以使用这些值来选择矩阵A的行,在我的情况下,这是一个5266 x 441矩阵?输出应为6 x 441矩阵。

1 个答案:

答案 0 :(得分:1)

虽然我没有看到你的列表之间的区别(为什么一个有逗号,另一个没有?),我认为你可以使用tf.gather函数来结束你想要的矩阵得到:https://www.tensorflow.org/api_docs/python/tf/gather