open3d.ml.torch.datasets.utils.create_3D_rotations#

open3d.ml.torch.datasets.utils.create_3D_rotations(axis, angle)#

Create rotation matrices from a list of axes and angles. Code from wikipedia on quaternions.

Parameters:

  • axis – float32[N, 3]

  • angle – float32[N,]

Returns:

float32[N, 3, 3]