读取多个.gz文件并以一个张量返回它

Question

我试图读取多个.gz文件，并以一个张量返回其内容，如下所示：

with ReadHelper('ark: gunzip -c /home/mnabih/kaldi/egs/timit/s5/exp/mono_ali/*.gz|') as reader:
    for key, b in reader:
        #print(type(b))
        c = torch.from_numpy(b)
        labels = torch.cat(c)

不幸的是，它给了我一个错误：参数“张量”（位置1）必须是张量的元组，而不是张量

有人能解释原因或解决方案吗

Answer 1

正如错误消息所解释的，c是一个张量。要使用torch.cat()，必须传递一组张量或一个列表。要解决您的问题，您可以使用：

temp = list()
for key, b in reader:
    temp.append(torch.from_numpy(b))
labels = torch.cat(temp)

有关更多信息，请查看the manual here

干杯