src/transformers/models/layoutlmv3/modeling_tf_layoutlmv3.py/TFLayoutLMv3PatchEmbeddings/call
class TFLayoutLMv3PatchEmbeddings:    def call(self, pixel_values: tf.Tensor) -> tf.Tensor:
        # When running on CPU, `tf.keras.layers.Conv2D` doesn't support `NCHW` format.
        # So change the input format from `NCHW` to `NHWC`.
        pixel_values = tf.transpose(pixel_values, perm=[0, 2, 3, 1])

        embeddings = self.proj(pixel_values)
        embeddings = tf.reshape(embeddings, (-1, self.num_patches, self.hidden_size))
        return embeddings