paddle里怎么计算两两之间的距离
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
切换版块