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:
[ [  [cos(ang(0)), -sin(ang(0))],  [sin(ang(0)), cos(ang(0))] ], [  [cos(ang(1)), -sin(ang(1))], [sin(ang(1)), cos(ang(1))] ],  .... ]