标签: python arrays tensorflow element
让我假设我有一个像张量
test = tf.constant([1,2,3],[4,5,6],[7,8,9])
我希望在这个张量中得到一些值(不是行或col)。 我怎样才能做到这一点?我想得到(2,3)这里的6的值。
我也想在多维数组中这样做。
'切片'或者其他张量函数不适合我,因为我想处理多维数组(不仅是2x2而且是2x3x4x10等)。
请帮助
答案 0 :(得分:0)
您想要的是tf.gather或tf.gather_nd