def forward()

in glide_text2im/clip/encoders.py [0:0]

• 4 lines of code
• 2 McCabe index (conditional complexity)

    def forward(self, x: torch.Tensor) -> torch.Tensor:
        if len(x.shape) != 2:
            raise ValueError()

        return F.embedding(x, self.w_voc) + self.w_pos[None, :, :]