如何有效地修改/制作成对距离矩阵？

Question

    x_norm = (x**2).sum(1).view(-1, 1)
    if y is not None:
        y_norm = (y**2).sum(1).view(1, -1)
    else:
        y = x
        y_norm = x_norm.view(1, -1)
    dist = (x_norm + y_norm - 2.0 * torch.mm(x, torch.transpose(y, 0, 1)))
    return dist

上面是用于计算x（M个点）和y（N个点）之间的成对距离矩阵（M*N）的代码。

当两点之间的距离大于特定值'T'时，我希望制作具有0元素的成对距离矩阵。

在这种情况下，我该怎么办？

谢谢

Answer 1

我想你在找torch.where:

new_dist = troch.where(dist > T, dist, 0.)