(X, Z=None)
| 59 | |
| 60 | |
| 61 | def pdist2_slow(X, Z=None): |
| 62 | if Z is None: Z = X |
| 63 | D = torch.zeros(X.size(0), X.size(2), Z.size(2)) |
| 64 | |
| 65 | for b in range(D.size(0)): |
| 66 | for i in range(D.size(1)): |
| 67 | for j in range(D.size(2)): |
| 68 | D[b, i, j] = torch.dist(X[b, :, i], Z[b, :, j]) |
| 69 | return D |
| 70 | |
| 71 | |
| 72 | if __name__ == "__main__": |