Hi @Paran0, this code does the work.
import torch

n = 10 #from 0 upto 9 degrees
thetas = np.arange(n, dtype = np.float32)

cosine = torch.from_numpy(np.cos(np.deg2rad(thetas)))
sine = torch.from_numpy(np.sin(np.deg2rad(thetas)))

batch = torch.zeros((n,2,2), dtype = cosine.dtype)

batch[:,0,0] = cosine
batch[:,0,1] = -sine
batch[:,1,0] = sine
batch[:,1,1] = cosine
print(batch.shape)