Hi guys, Let’s say we want to generate Rotation Matrices from Batch of angles ( Bx1), we could use the torch.cos() and torch.sin() to convert (Bx1) angles to (Bx1) tensors of cos and sin.

How I can stack the given angles to produce the batched version of Rotation Matrices.
Here is an example: