tf.extract_image_patches用于3D图像

Question

The documentation of tf.extract_image_patches

仅适用于2D图像，是否可以扩展为3D图像，这对SSIM丢失功能的实现有用吗？

我找不到源代码。有一个类似的函数skimage.util.view_as_windows，但是，当我尝试使用此函数将tensorflow作为keras中的后端时，会出现错误。从numpy数组到张量的过渡让我很困惑。