The patches would have the shape [n_patches, patch_height, patch_width] or [n_patches, patch_height, patch_width, n_channels] so you could access the center pixels by indexing dim1 with patch_height//2 and dim2 with patch_width//2.