有沒有辦法在 3D 矩陣的兩個維度上并行計算矩陣運算……？例如，我想計算每個sig級別的(3x2) 矩陣的偽逆矩陣，G這樣得到的矩陣將是 (2x3x5)，其sig維度不包括在計算逆矩陣中；倒數(shù)僅適用于每個級別的row和col維度。sig理想情況下，我想以下面類似的方式實現(xiàn)這一點：G = xr.DataArray(np.random.rand(3,2,5), dims=['row','col','sig'])pinvG = xr.apply_ufunc(np.linalg.pinv, G.chunk({'sig':1}), output_core_dims=[['j','i']], # input_core_dims=[['sig'],['sig']],# exclude_dims={'sig','sig'}, dask='parallelized', output_dtypes=[float], output_sizes={'j':2,'i':3}).compute()
1 回答
慕的地6264312
TA貢獻1817條經(jīng)驗 獲得超6個贊
我能夠通過使用來做我想做的事dask.array.map_blocks：
pinvG = dsar.map_blocks(np.linalg.pinv,G.transpose('sig','row','col').chunk({'sig':1}).data)添加回答
舉報
0/150
提交
取消