我试图填充一批列表,这样我就得到了一个矩形张量。上下文是我想在不相等的特征向量上运行批训练。我相信在tensorflow中应该有一个简单的实现方法,但是我找不到。你知道吗
在使用feed_dict
传递特征向量之前,我已经尝试过对特征向量进行“预填充”,而且效果很好。但是,我想把它作为一个本机Tensorflow操作来执行,因为python中的预填充可能比在图中执行要慢。你知道吗
def pad(train_features_list):
max_seq_len = max(len(features) for features in train_features_list)
for k, _ in enumerate(train_features_list):
pads_to_add = max_seq_len - len(train_features_list[k])
train_features_list[k].extend(['<PAD>'] * pads_to_add)
目前没有回答
相关问题 更多 >
编程相关推荐