以十为单位填充一批不均匀/不等的数组

2024-06-17 10:37:53 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图填充一批列表,这样我就得到了一个矩形张量。上下文是我想在不相等的特征向量上运行批训练。我相信在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)

Tags: toinadd列表forlentensorflowtrain