julius/prototorch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add alias for squared_euclidean_distance

Browse Source
This commit is contained in:
blackfly 2020-04-14 19:53:26 +02:00
parent a3548e0ddd
commit 88cbe0a126
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
prototorch/functions/distances.py
View File

@ -69,3 +69,7 @@ def lomega_distance(x, y, omegas):
distances = torch.sum(differences_squared, dim=2) distances = torch.sum(differences_squared, dim=2)
distances = distances.permute(1, 0) distances = distances.permute(1, 0)
return distances return distances
# Aliases
sed = squared_euclidean_distance