我正在尝试使用 numpy 的 cumprod 的功能,但用于矩阵。有一个充满齐次变换的矩阵(为了简单起见,这里是随机的)我想使用 numpy 的矩阵乘法来累积变换。
n = 1000
data = np.random([n, 4,4])
dataAcc = cumProd(data, np.matmul)
dataAcc 应包含以下内容:
dataAcc[0,:,:] = data[0,:,:]
dataAcc[1,:,:] = data[0,:,:] @ data[1,:,:] = dataAcc[0,:,:] @ data[1,:,:]
dataAcc[2,:,:] = dataAcc[1,:,:] @ data[2,:,:]
...
有没有办法使用快速的 numpy 函数或类似的函数来做到这一点?我不想使用循环...
干杯