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

Question

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

我有一个 2D 张量之类的东西x => [[0,1],[1,2],[2,3]]，我想y用它来切片一个 3D 张量。例如y[x[:,0], x[:,1], :]，但它不起作用。以下是错误：

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

任何人都可以请帮助！谢谢

标签： pythonnumpytensorflow

Answer 1

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

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