Как тензорировать потерю нескольких 3D-точек

Question

У меня есть тензор основных истинных значений трехмерных точек G = [18000x3] и вывод из моей сети того же размера O = [18000x3].

Мне нужно вычислить убыток, чтобыу меня есть квадратный корень расстояния между каждой трехмерной точкой, сложенный по всем ключевым точкам и нормализованный за 18000. Как мне эффективно написать это?

Answer 1

Просто напишите предложенное вами выражение, используя векторизованные операции, предоставляемые PyTorch.В этом случае

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

Выезд pow, sum, sqrt и mean.