torch.cdist 是PyTorch 中用于計(jì)算**兩個(gè)張量之間的成對(duì)距離(pairwise distance)**的函數(shù),常用于點(diǎn)云處理、圖神經(jīng)網(wǎng)絡(luò)、相似性度量等場(chǎng)景。 基本語法 1 torch.cdist(x1, x2, p=2.0) 參數(shù)說明: 參數(shù)說明 x1 一個(gè)形狀為 [B, M, D] 或[M, D] 的張量,表示一組點(diǎn)。 x2 一個(gè)形狀為 [B, N, D
www.dbjr.com.cn/python/342092v...htm 2025-6-5