starling.inference.generation.symmetrize_distance_map
- symmetrize_distance_map(dist_map)[source]
Symmetrize a distance map by replacing the lower triangle with the upper triangle values.
- Parameters:
dist_map (torch.Tensor) – A 2D tensor representing the distance map.
- Returns:
A symmetrized distance map.
- Return type: