coding小技巧记录
程序员文章站
2023-12-30 16:30:22
...
# 创建空间正方体，返回坐标
coords = np.mgrid[:resX, :resY, :resZ]
# 快速对3D点云进行旋转平移操作
pts = torch.baddbmm(trans, rot, points) # [B, 3, N]