(self, x)
| 24 | self.attention_c = nn.Linear(D, n_classes) |
| 25 | |
| 26 | def forward(self, x): |
| 27 | a = self.attention_a(x) |
| 28 | b = self.attention_b(x) |
| 29 | A = a.mul(b) |
| 30 | A = self.attention_c(A) # N x n_classes |
| 31 | return A |
| 32 | |
| 33 | class Attn_Modality_Gated(nn.Module): |
| 34 | # Adapted from https://github.com/mahmoodlab/PORPOISE |
nothing calls this directly
no outgoing calls
no test coverage detected