فاصله‌ی دوبه‌دو با پخش (broadcasting)

A ماتریس $n \times p$ و B ماتریس $m \times p$ است. تابع sq_dists(A, B) ماتریس $n \times m$ مربع فاصله‌ی اقلیدسی هر ردیف A تا هر ردیف B را برمی‌گرداند:

بدون هیچ حلقه‌ای: یا با پخش — A[:, None, :] - B[None, :, :] آرایه‌ای $n \times m \times p$ می‌سازد — یا با اتحاد $\lVert a - b\rVert^2 = \lVert a\rVert^2 + \lVert b\rVert^2 - 2\,a \cdot b$ و یک ضرب ماتریسی. همین کد با torch به‌جای np روی GPU هم اجرا می‌شود.