假设我有一个inputshape张量100x1、另一个inplaceshape张量20x1和一个index_tensorshape 100x1。代表我想要插入值index_tensor的位置。只有20 个 True 值,其余值为 False。我尝试在下面解释所需的操作。
inputinplaceindex_tensor
使用tensorflow如何实现这个操作。
assign操作仅适用于tf.Variable我想将其应用于 的输出tf.nn.rnn。
我读到可以使用,tf.scatter_nd但它需要inplace和index_tensor具有相同的形状。
我想使用它的原因是我从 rnn 获得输出,然后我从中提取一些值并将它们输入到某个致密层,而来自致密层的输出,我想插入回从 rnn 获得的原始张量中手术。由于某些原因,我不想对 rnn 的整个输出应用密集层操作,如果我不将密集层的结果插入到 rnn 的输出中,那么密集层就没用了。
任何建议将不胜感激。