(self, state)
| 61 | self.bidirectional = bidirectional |
| 62 | |
| 63 | def __setstate__(self, state): |
| 64 | if 'activation' not in state: |
| 65 | state['activation'] = F.relu |
| 66 | super(TransformerEncoderLayer, self).__setstate__(state) |
| 67 | |
| 68 | def forward(self, src, src_mask=None, src_key_padding_mask=None): |
| 69 | # type: (Tensor, Optional[Tensor], Optional[Tensor]) -> Tensor |
nothing calls this directly
no outgoing calls
no test coverage detected