starling.inference.constraints.symmetrize_distance_maps
- symmetrize_distance_maps(dist_maps: Tensor) Tensor[source]
Symmetrize a batch of distance maps in PyTorch.
- Parameters:
dist_maps (torch.Tensor) – Tensor of shape (B, N, N) representing pairwise distances.
- Returns:
Symmetrized distance maps with zero diagonal.
- Return type: