(self, step=-1)
| 135 | """Take a specific time step from 3D tensor.""" |
| 136 | |
| 137 | def __init__(self, step=-1): |
| 138 | self.step = step |
| 139 | |
| 140 | def forward_pass(self, x): |
| 141 | return x[:, self.step, :] |
nothing calls this directly
no outgoing calls
no test coverage detected