LieTorch:相切空间反向传播
介绍
LieTorch库将PyTorch概括为3D变换组。 正如torch.Tensor是标量元素的多维矩阵一样, lietorch.SE3是SE3元素的多维矩阵。 我们支持常见的张量操纵,例如索引,重塑和广播。 可以将组运算组成计算图,并在每个元素的切线空间中自动进行反向传播。 有关更多详细信息,请参见我们的论文:
扎卡里·泰德(Jachary Teed)和贾登(Ceng Deng),CVPR 2021
@inproceedings{teed2021tangent,
title={Tangent Space Backpropagation for 3D Transformation Groups},
author={Teed, Zachary and Deng, Jia},
booktitle={Conference on Comput
2021-08-23 10:21:24
5.82MB
Cuda
1