pytorch中對於tensor的一些騷操作

使用環境:pytorch1.3.1;python3.6 最近在讀pytorch官方實現的Faster RCNN代碼時,發現了一些非常規的tensor操作,爲了方便理解,在此記錄下: 1. (a[:, None] * b[None, :]).view(-1) 這是在計算anchor的k值出現的操作,其中a爲一維向量[a1, a2, ..., an],b也爲一維向量[b1, b2, ..., bm]。
相關文章
相關標籤/搜索