Model对象嵌套list赋值方式（备忘）

要想给一个嵌套的列表赋值，可以使用类似于下面的方式：

import torch

model = torch.nn.Sequential(
    torch.nn.Linear(10, 10),
    torch.nn.ReLU(),
    torch.nn.Linear(10, 5),
    torch.nn.ReLU(),
    torch.nn.Linear(5, 1)
)

data = [
    torch.randn(10),
    torch.randn(10),
    torch.randn(10),
    torch.randn(10),
    torch.randn(10)
]

output = []
for x in data:
    output.append(model(x))

print(output)

在这个例子中，我们首先定义了一个简单的神经网络模型 `model`，它包含了三个线性层和两个 ReLU 激活函数。然后我们定义了一个包含 5 个元素的列表 `data`，每个元素都是一个长度为 10 的张量。最后我们使用一个 for 循环遍历 `data`，将每个张量输入到模型中，并将输出添加到列表 `output` 中。最终输出的结果是一个包含 5 个元素的列表，每个元素都是一个标量（即模型的输出）。