(self, f: torch.FloatTensor, zq: torch.FloatTensor)
| 1690 | self.conv_b = nn.Conv2d(zq_channels, f_channels, kernel_size=1, stride=1, padding=0) |
| 1691 | |
| 1692 | def forward(self, f: torch.FloatTensor, zq: torch.FloatTensor) -> torch.FloatTensor: |
| 1693 | f_size = f.shape[-2:] |
| 1694 | zq = F.interpolate(zq, size=f_size, mode="nearest") |
| 1695 | norm_f = self.norm_layer(f) |
| 1696 | new_f = norm_f * self.conv_y(zq) + self.conv_b(zq) |
| 1697 | return new_f |
| 1698 | |
| 1699 | |
| 1700 | ## Deprecated |
nothing calls this directly
no outgoing calls
no test coverage detected