paddle里怎么计算两两之间的距离
玉扉洞主 发布于2021-08 浏览:1358 回复:0
1
收藏

我有两个tensor,A是 B * N * D,B是 B * M * D,想要得到每个batch里A中所有点和B中所有点的欧氏距离,也就是得到一个B * N * M的矩阵,其中B是batch_size,N和M分别是点的个数,D是特征维度(假设是2d坐标,D=2)。N*M的矩阵里的结果是N和M个点的两两距离。这在scipy或者pytorch里可以直接用 cdist(A, B) 得到,但是好像在paddle里没有对应的函数?

如果通过循环b次来计算的话,又丢失了并行性,请问有什么对应的解决方案吗?

收藏
点赞
1
个赞
TOP
切换版块