分裂张量

Question

让T为形状[n,f]的张量，代表批次。现在，我想将T切片到m张量axis=0。 m的值取决于当前批次。我有另一个形状I的张量[m,2]，它存储了一对指数，用于指示切片的位置。

我不确定如何＆＃34;迭代＆＃34;在指数上申请tf.slice。有什么想法吗？

这可以通过tf.scan以某种方式实现吗？

Answer 1

我想你正在寻找split功能。