Pytorch 如何连接like列表

我不想在类似“列表理解”之前定义输入列表和输出列表 请告诉我在这种情况下怎么办

多谢各位

e.g.
inputs = torch.rand(1,5)
weights = torch.rand(5,5)

def training(self,inputs,weights,steps=50):
    input_list=torch.empty(1,5)
    output_list=torch.empty(5,5)

    for i in range(steps):
        outputs = torch.matmul(inputs,weights)

        input_list = torch.cat([input_list,inputs], axis=0)
        output_list = torch.cat([output_list,outputs], axis=0)

        inputs = outputs

    return inputs,input_list,output_list

他们 输入列表为 张量（[[1.8367e-36,0.0000e+00,3.3631e-44,0.0000e+00，nan]， [3.9109e-01、8.9431e-01、6.8890e-01、8.3894e-01、1.7802e-01]）

---

我不需要第一排。在这种情况下如何连接？

这个问题由专业编程人员解决

像这样,

input_list=torch.tensor([])
output_list=torch.tensor([])

参考文献