Tf.extract_image_patches for 3D images

Tf.extract_image_patches documentation

For 2D image only, it can be expanded to 3D images, what is useful for realizing SSIM loss function?

I cannot find the source code. There is a similar function skimage.util.view_as_windows

, however, when I try to use this function with tensorflow as a backend in keras, errors are thrown. Going from numpy array to tensor confused me a lot.