假设我有一个形状为
如何使用tensorflow实现此操作。
我想使用这个的原因是,我从rnn得到一个输出,然后从中提取一些值并将它们馈送到某些密集层,我希望将这个密集层的输出插回到我从rnn操作中获得的原始张量中。由于某些原因,我不想对整个rnn输出应用密集层操作,如果我不将密集层的结果插入到rnn输出中,那么密集层就有点无用了。
100x1
的张量input
,另一个形状为20x1
的张量inplace
和一个形状为100x1
的index_tensor
。 index_tensor
表示我想要将inplace
的值插入到input
的位置。 index_tensor
只有20个True值,其余值为False。 我尝试在下面解释所需的操作。
![enter image description here](https://istack.dev59.com/ogX23.webp)
assign
操作仅适用于tf.Variable
,而我希望将其应用于tf.nn.rnn
的输出。
我读到可以使用tf.scatter_nd
,但它要求inplace
和index_tensor
具有相同的形状。我想使用这个的原因是,我从rnn得到一个输出,然后从中提取一些值并将它们馈送到某些密集层,我希望将这个密集层的输出插回到我从rnn操作中获得的原始张量中。由于某些原因,我不想对整个rnn输出应用密集层操作,如果我不将密集层的结果插入到rnn输出中,那么密集层就有点无用了。
任何建议将不胜感激。