作者:
odanaga (PixiyON)
2018-02-19 19:55:29新年快樂
小弟python新手想問個numpy問題
今天我有一個shape = (40,3)的A
還有一個shape = (40,1)的B
利用np.multiply可以乘出一個shape = (40,3)的矩陣
若我今天有一個C.shape = (260,40,1)
我要怎樣才能(不用for迴圈)乘出一個(260,40,3)的矩陣呢?
謝謝
一樣直接np.multiply(A, C)就可了吧 broadcasting不然也可以用A[np.newaxis, :, :]先墊一個維度再做multiply，這樣就和第一個例子差不多了c.shape=(260, 40)時不行是因為broadcasting的規則是把大小是1的那個維度拉到跟另外一個陣列一樣，那個時候A和C完全沒有大小是1的維度所以才無法成功