theano broadcasting

當咱們使用函數對兩個數組進行計算時,函數會對這兩個數組的對應元素進行計算,所以它要求這兩個數組有相同的大小(shape相同)。若是兩個數組的shape不一樣的話,會進行以下的廣播(broadcasting)處理:數組

  • 讓全部輸入數組都向其中shape最長的數組看齊,shape中不足的部分都經過在前面加1補齊
  • 輸出數組的shape是輸入數組shape的各個軸上的最大值
  • 若是輸入數組的某個軸和輸出數組的對應軸的長度相同或者其長度爲1時,這個數組可以用來計算,不然出錯
  • 當輸入數組的某個軸的長度爲1時,沿着此軸運算時都用此軸上的第一組值
相關文章
相關標籤/搜索