廣播法則(broadcast)

  廣播法則(broadcast)是科學運算中經常使用的一個技巧,它在快速執行向量化的同時不會佔用額外的內存/顯存。 Numpy的廣播法則(3條)定義如下: 讓所有輸入數組都向其中shape最長的數組看齊,shape中不足的部分通過在前面加1補齊 兩個數組要麼在某一個維度的長度一致,要麼其中一個爲1,否則不能計算 當輸入數組的某個維度的長度爲1時,計算時沿此維度複製擴充成一樣的形狀 PyTorch
相關文章
相關標籤/搜索