tensorflow - 根据存储在每个批次的另一个张量中的索引，在 tensorflow 中更新批次中的张量切片

Question

给定一个形状为 [None,#number of timesteps] 的布尔掩码张量（全部设置为 false）。还有另一个张量，其中每个批次都存储了索引。索引范围是 (0,#number of timesteps)。它的形状是[None,1]。对于每个批次单独使用索引张量，应该将原始布尔掩码的一部分更新为 True。

例如，假设掩码为 [[False,False,False,False],[False,False,False,False]]，索引张量为 [2,3]。输出应该是 [[True,True,False,False],[True,True,True,False]]。即，在第一批 [:2] 中设置为真，在第二批中 [:3] 设置为真。

如何在不使用会话调用和一次所有批次的情况下在 tensorflow 中实现上述目标？

标签： tensorflowslicebatch-processingtensorassign