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

问题描述 投票:2回答:1

我正在尝试读取多个.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)

不幸的是,它给了我这个错误cat():参数'张量'(位置1)必须是张量的元组,而不是张量

任何人都可以解释原因或解决方法

python numpy pytorch torch
1个回答
1
投票

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

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

有关更多,您可以检查the manual here

欢呼声

© www.soinside.com 2019 - 2024. All rights reserved.