src/transformers/models/vilt/modeling_vilt.py/ViltPatchEmbeddings/forward
class ViltPatchEmbeddings:    def forward(self, pixel_values):
        batch_size, num_channels, height, width = pixel_values.shape
        if num_channels != self.num_channels:
            raise ValueError(
                "Make sure that the channel dimension of the pixel values match with the one set in the configuration."
            )
        x = self.projection(pixel_values)
        return x