(self, output_size)
| 821 | """ |
| 822 | |
| 823 | def __init__(self, output_size) -> None: |
| 824 | super().__init__() |
| 825 | assert output_size is not None, "'output_size' cannot be NoneType" |
| 826 | self.output_size = _triple(output_size) |
| 827 | |
| 828 | def forward(self, x): |
| 829 | assert ( |