myoverse.models.components.losses.EuclideanDistance.forward

EuclideanDistance.forward(prediction, ground_truth)

Compute the mean Euclidean distance loss.

Parameters:

• prediction (torch.Tensor) – Predicted joint positions, shape (batch, n_joints * n_dims).

• ground_truth (torch.Tensor) – Ground truth joint positions, shape (batch, n_joints * n_dims).

Returns:

Scalar loss value.

Return type:

torch.Tensor