(self)
| 433 | |
| 434 | class Simple(Module): |
| 435 | def __init__(self): |
| 436 | super().__init__() |
| 437 | self.conv0 = Conv2d(1, 1, kernel_size=3, bias=False) |
| 438 | self.conv1 = Conv2d(1, 1, kernel_size=3, bias=False) |
| 439 | self.conv1.weight = self.conv0.weight |
| 440 | |
| 441 | def forward(self, inputs): |
| 442 | x = self.conv0(inputs) |