如何张紧多个3D关键点的丢失

Question

我有一个G = [18000x3]的3D点的地面真实值张量，并且我的网络的输出大小为O = [18000x3]。

我需要计算一个损失，以便基本上获得每个3D点之间的距离的平方根，对所有关键点求和并在18000上进行归一化。

Answer 1

只需使用PyTorch提供的向量化操作编写您建议的表达式。在这种情况下

loss = (O - G).pow(2).sum(axis=1).sqrt().mean()

签出pow，sum，sqrt和mean。