Method
__init__
(self, d_model, d_model_out, scale_factor, resolution=None, view_num=1)
Source from the content-addressed store, hash-verified
| 518 | Upsampling block. |
| 519 | """ |
| 520 | def __init__(self, d_model, d_model_out, scale_factor, resolution=None, view_num=1): |
| 521 | super().__init__() |
| 522 | |
| 523 | self.scale_factor = scale_factor |
| 524 | self.d_model_out = d_model_out |
| 525 | self.residual_fc = nn.Linear(d_model, d_model_out * (scale_factor**2), bias=True) |
| 526 | self.pixelshuffle = nn.PixelShuffle(scale_factor) |
| 527 | |
| 528 | self.resolution = resolution |
| 529 | |
| 530 | self.view_num = view_num |
| 531 | |
| 532 | def forward(self, x): |
| 533 | x = self.residual_fc(x) |
Callers
nothing calls this directly
Tested by
no test coverage detected