如何在给定的索引中插入索引？

Question

我有一个12 x 10大小的矩阵和一个文档长度列表=[4,3,5]。前4行属于一个文档的句子，第3行属于其他文档，最后5行属于其他文档。我需要用下面的方法得到一个新的矩阵或张量，大小为15×10。在

我需要插入一行0，以这样的方式，这里的最大doc长度是5。因此，必须在第5个索引中插入1行零（因为第一个文档的长度是4）。然后在索引9和10处有2行零（第二个文档长度是3），依此类推。如果这个问题令人困惑，请告诉我。在

我已经给GRU或LSTM提供了12 x 10，但必须是3d格式。为了得到一个合适的3D，我需要把12 X 10转换成15 X 10。有什么帮助吗？在

Answer 1

这里有一个方法。首先将12x10张量分解为一个大小为12的列表。这是使用tf.unstack()函数完成的。在

# Assume `words` is my 12x10 tensor
tensors = tf.unstack(words, 12, axis=0) 

之后，可以在索引处的列表中插入tf.zeros()张量。为了简单起见，我只在索引4中插入一个张量。在

^{pr2}$

最后，tf.stack()重新构建列表。在

words = tf.stack(tensors, axis=0)