gpt4 book ai didi

python - 使用 numpy/pytorch 广播计算矩阵乘积的轨迹

转载 作者:行者123 更新时间:2023-12-05 07:34:44 24 4
gpt4 key购买 nike

令 A 为 (nxm) 矩阵,M 为 (mxm) 矩阵。为矩阵的踪迹编写 tr(),我需要计算 tr(AM(A^T))。然而,最后的跟踪操作丢弃了大部分计算。我可以使用 numpy 或 pytorch 的广播规则来仅计算 AM(A^T) 的必要对角线吗?

更新:这是我在 PyTorch 中计算对角线的解决方案:

torch.sum(torch.sum(A.t()[:,None,:]*M[:,:,None],0)*A.t(),0)

最佳答案

您必须至少计算两个矩阵乘积中的一个。随后您可以使用此处的答案之一:What is the best way to compute the trace of a matrix product in numpy?

关于python - 使用 numpy/pytorch 广播计算矩阵乘积的轨迹,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49992174/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com