(self, x)
| 51 | self.dim = 2 if batch else 1 |
| 52 | |
| 53 | def forward(self, x): |
| 54 | x = self.affine1(x) |
| 55 | x = self.dropout(x) |
| 56 | x = F.relu(x) |
| 57 | action_scores = self.affine2(x) |
| 58 | return F.softmax(action_scores, dim=self.dim) |
| 59 | |
| 60 | |
| 61 | class Observer: |
nothing calls this directly
no outgoing calls
no test coverage detected