def METHOD_NAME(self, x):
    B = x.shape[0]
    x = self.patch_embed(x)
    for i, blk in enumerate(self.blocks):
        x = blk(x)
    x = self.norm(x)
    x = x.mean(dim=1).reshape(B, 1, -1)
    return x[:, 0]