numpy和Pytorch的廣播機制

廣播（broadcast）規則 對於兩個數組，對於它們的維度元組倒序看，當二者在當前維度的尺寸相等或者有一個爲1，則在當前維度兼容，只須要將尺寸爲1的按照另外一個較大的尺寸在該維度延展。若是某一個數組沒有對應維度時則用1來補。高維度矩陣廣播也能夠按照這個來分析。css 示例 Pytorch中的視圖方法(view) def squared_loss(y_hat, y): return (y_hat

>>阅读原文<<