(self, x: torch.Tensor)
| 52 | self.posemb_version = posemb_version |
| 53 | |
| 54 | def forward(self, x: torch.Tensor): |
| 55 | if self.posemb_version is not None: |
| 56 | x = encode_position(self.posemb_version, position=x) |
| 57 | return super().forward(x) |
| 58 | |
| 59 | |
| 60 | class MultiviewPoseEmbedding(nn.Conv2d): |
no test coverage detected