(self, dim_in: int, dim_out: int, bias: bool = True)
| 158 | """ |
| 159 | |
| 160 | def __init__(self, dim_in: int, dim_out: int, bias: bool = True): |
| 161 | super().__init__() |
| 162 | self.proj = nn.Linear(dim_in, dim_out, bias=bias) |
| 163 | |
| 164 | def forward(self, x: torch.Tensor) -> torch.Tensor: |
| 165 | x = self.proj(x) |