标签： pytorch
我计划实现自定义功能。
但是，如果输入张量是矢量，而输出张量是矢量。 结果将是雅可比矩阵。
例如，该功能是计算射线-球面交点。输出是位置。我需要使用CUDA做到这一点。但是，我可能需要在正向传递过程中保存一个雅可比矩阵。有更有效的方法吗？