(self, x, f, mask)
| 222 | self.alpha = alpha |
| 223 | |
| 224 | def forward(self, x, f, mask): |
| 225 | out = self.feed_forward(x) |
| 226 | out = self.alpha * self.att_layer(self.instance_norm(out), f, mask) + out |
| 227 | out = self.conv_1x1(out) |
| 228 | out = self.dropout(out) |
| 229 | return (x + out) * mask[:, 0:1, :] |
| 230 | |
| 231 | |
| 232 | class PositionalEncoding(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected