如何利用 kornia 旋转图片以及旋转点坐标

// 得到旋转矩阵
M: torch.tensor = kornia.get_rotation_matrix2d(center, angle, scale)  # (batch_size, 2, 3)

// 得到旋转后的图片
x_warped: torch.tensor = kornia.warp_affine(x.float(), M.to(x.device), dsize=(h, w))

// 原图中某像素索引
origin_index = torch.tensor([[[a, b]]]).float()  # （B,N,D）

zzo = torch.tensor([0,0,1])
zzo = zzo.repeat(M.shape[0],1,1)
// 完整的旋转矩阵
M2 = torch.cat([M,zzo],dim=1)  # (batch_size, 3, 3)
// 旋转后图像中对应像素的索引
index = kornia.geometry.transform_points(M2, origin_index)