Pytorch 无法分配'；torch.nn.modules.container.Sequential'；作为参数

我一直在遵循这个方法
（）在

forward

函数中动态分配参数

然而，我的参数不仅仅是一个单权张量，而是

nn.Sequential

当我执行以下操作时：

class MyModule(nn.Module):
    def __init__(self):
        # you need to register the parameter names earlier
        self.register_parameter('W_di', None)

    def forward(self, input):
        if self.W_di is None:
            self.W_di = nn.Sequential(
                nn.Linear(mL_n * 2, 1024),
                nn.ReLU(),
                nn.Linear(1024, self.hS)).to(device)

我得到以下错误

 TypeError: cannot assign 'torch.nn.modules.container.Sequential' as parameter 'W_di' (torch.nn.Parameter or None expected)

是否有任何方法可以将

nn.Sequential

注册为一个整体参数？谢谢