(self, x)
| 530 | self.view_num = view_num |
| 531 | |
| 532 | def forward(self, x): |
| 533 | x = self.residual_fc(x) |
| 534 | bs, l, c = x.shape |
| 535 | x = x.reshape(bs * self.view_num, self.resolution[0], self.resolution[1], c).permute(0, 3, 1, 2) |
| 536 | x = self.pixelshuffle(x) |
| 537 | x = x.permute(0, 2, 3, 1).reshape(bs, self.view_num, self.resolution[0]*self.scale_factor, self.resolution[1]*self.scale_factor, self.d_model_out) |
| 538 | x = x.reshape(bs, -1, self.d_model_out) |
| 539 | return x |
| 540 | |
| 541 | class GaussianUpsampler(nn.Module): |
| 542 | """ |
nothing calls this directly
no outgoing calls
no test coverage detected