通过另一个变量Tensorflow切片变量

Question

我有一个项目的numpy代码，并想将其转换为tensorflow。

我有一个2D张量，例如x => [[0,1],[1,2],[2,3]]等，我想使用它来切片3D张量y。例如y[x[:,0], x[:,1], :]，但无效。以下是错误：

ValueError: Shape must be rank - but is rank - for 'strided_slice_?' (op: 'StridedSlice') with input shapes: [-], [-], [-], [-].

任何人都可以帮忙！ 谢谢

Answer 1

您需要标量才能索引到y，而不是1+级的张量。

尝试y[x[0, 0], x[0, 1], :]进行快速测试。