2022-08-24
閱讀量:
991
廣播機制
當我們使用通用函數(shù)對兩個數(shù)組進行計算時,通用函數(shù)會對這兩個數(shù)組的對應元素進行計算,因此它要求這兩個數(shù)組shape相同。如果shape不同的話,會進行如下的廣播(broadcasting)處理
讓所有輸入數(shù)組都向其中shape最長的數(shù)組看齊,shape中不足的部分都通過在前面加1補齊
輸出數(shù)組的shape是輸入數(shù)組shape的各個軸上的最大值
如果輸入數(shù)組的某個軸和輸出數(shù)組的對應軸的長度相同或者其長度為1時,這個數(shù)組能夠用來計算,否則出錯
當輸入數(shù)組的某個軸的長度為1時,沿著此軸運算時都用此軸上的第一組值






評論(0)


暫無數(shù)據(jù)
推薦帖子
0條評論
0條評論
0條評論
0條評論