(self, x)
| 248 | h, w = output_size |
| 249 | |
| 250 | def forward(self, x): |
| 251 | feat = self.embedding(x) |
| 252 | b, n, c = feat.size() |
| 253 | feat = feat.permute(0, 2, 1) |
| 254 | feat = self.to_patch(feat) |
| 255 | |
| 256 | return feat |
| 257 | |
| 258 | class MultiHeadedAttention(nn.Module): |
| 259 | """ |
nothing calls this directly
no outgoing calls
no test coverage detected