可以通过以下步骤实现:
import cupy as cp
input_matrix = cp.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
distance_matrix = cp.sqrt(((input_matrix[:, None, :] - input_matrix) ** 2).sum(axis=2))
这里使用广播机制将输入矩阵进行扩展,然后计算每个元素的欧几里得距离,并使用cupy的sqrt函数对结果进行开方。
距离矩阵是一个对称矩阵,其中每个元素表示矩阵中对应两行之间的距离。对角线上的元素为0,表示每行与自身的距离为0。
优势:
应用场景:
推荐的腾讯云相关产品:
更多关于腾讯云产品的信息,请访问腾讯云官方网站:https://cloud.tencent.com/
领取专属 10元无门槛券
手把手带您无忧上云