如何访问张量中的每个元素?

时间:2017-07-25 11:07:18

标签: python arrays tensorflow element

让我假设我有一个像张量

的张量
test = tf.constant([1,2,3],[4,5,6],[7,8,9])

我希望在这个张量中得到一些值(不是行或col)。 我怎样才能做到这一点?我想得到(2,3)这里的6的值。

我也想在多维数组中这样做。

'切片'或者其他张量函数不适合我,因为我想处理多维数组(不仅是2x2而且是2x3x4x10等)。

请帮助

1 个答案:

答案 0 :(得分:0)

您想要的是tf.gathertf.gather_nd