question about the 3D rotation order and the Fourier transformation matrix

[HaojHuang]

Hi,

It seems torch.nn.functional.affine_grid changes the rotation result due to its 'zyx' indexing mechanism. Could I know what is the 3D rotation rules of the GeometricTensor? To be specific, when calculating the Fourier transform matrix, will the function FourierPointwise change the fed rotation matrixes following 'zyx' index automatically? If so, is it possible to get the Fourier transformation matrix without changing the rotation order?

Thanks,
Haojie