(self, x)
| 182 | self.linear_out = nn.Linear(in_dim if num_layers == 1 else inner_dim, out_dim) if num_layers >= 1 else nn.Identity() |
| 183 | |
| 184 | def forward(self, x): |
| 185 | x = self.linear_hidden(x) |
| 186 | x = self.linear_out(x) |
| 187 | |
| 188 | return x |
nothing calls this directly
no outgoing calls
no test coverage detected