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

时间:2020-07-09 15:50:43

标签: tensorflow slice batch-processing tensor assign

给出一个形状为[None,#timesteps]的布尔掩码张量(全部设置为false)。也有另一个张量,其中每个批次都存储有索引。索引范围是(0,#timesteps)。其形状为[None,1]。对于每个批次,分别使用索引张量将原始布尔掩码的一个切片更新为True。

例如,假设掩码为[[False,False,False,False],[False,False,False,False]],索引张量为[2,3]。输出应为[[True,True,False,False],[True,True,True,False]]。 也就是说,第一批[:2]设置为true,第二批[:3]设置为true。

如何在不使用会话调用且一次完成所有批次的情况下在tensorflow中实现以上目标?

0 个答案:

没有答案