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]