如何使用张量流旋转3D图像？

Question

我有一个图像张量：[批处理，高度，宽度，深度，3]和一个角度张量：[angleX，angleY，angleZ]。如何使用角度旋转图像？ tf.contrib.image.rotate仅适用于2D图像。

顺便说一句，我可以使用“ Y旋转（X，角度）”作为损失函数吗？我不确定tensorflow是否可以计算旋转函数的梯度。

Answer 1

github中有一个对旋转有用的代码。但是，它仅适用于2D图像。