Tensorflow tf.extract_image_patches for 3D images

Can someone provide an implementation of tf.extract_image_patches , but this works with 3D images (ideally n-dimensional tensors) which I was thinking about using tf.strided_slice , but I can't figure out how to use it without looping over tensor...