Clean up prototorch/functions/distances.py

This commit is contained in:
blackfly 2020-04-08 22:44:02 +02:00
parent bde408a80e
commit 7d5ab81dbf

View File

@ -33,13 +33,6 @@ def lpnorm_distance(x, y, p):
Expected dimension of x is 2. Expected dimension of x is 2.
Expected dimension of y is 2. Expected dimension of y is 2.
""" """
# # DEPRECATED in favor of torch.cdist
# expanded_x = x.unsqueeze(dim=1)
# batchwise_difference = y - expanded_x
# differences_raised = torch.pow(batchwise_difference, p)
# distances_raised = torch.sum(differences_raised, axis=2)
# distances = torch.pow(distances_raised, 1.0 / p)
# return distances
distances = torch.cdist(x, y, p=p) distances = torch.cdist(x, y, p=p)
return distances return distances