argmax of frame activations Args: torch.Tensor hs_pad: 3d tensor (B, Tmax, eprojs) Returns: torch.Tensor: argmax applied 2d tensor (B, Tmax)
(self, hs_pad)
| 50 | return F.log_softmax(self.ctc_lo(hs_pad), dim=2) |
| 51 | |
| 52 | def argmax(self, hs_pad): |
| 53 | """argmax of frame activations |
| 54 | |
| 55 | Args: |
| 56 | torch.Tensor hs_pad: 3d tensor (B, Tmax, eprojs) |
| 57 | Returns: |
| 58 | torch.Tensor: argmax applied 2d tensor (B, Tmax) |
| 59 | """ |
| 60 | return torch.argmax(self.ctc_lo(hs_pad), dim=2) |
no outgoing calls
no test coverage detected