import torch
import roma

# Example Euler angles (batch_size x 3)
euler_angles = torch.tensor([[0.1, 0.2, 0.3]])

# Convert to rotation matrix
rotation_matrix = roma.euler_to_rotmat(angles=euler_angles, convention='xyz')

rotation_6d = roma.rotmat_to_rot6d(rotation_matrix)