gpt4 book ai didi

python - 使用 einsum 的矩阵乘法

转载 作者:太空宇宙 更新时间:2023-11-04 05:13:02 25 4
gpt4 key购买 nike

<分区>

我想将两个具有以下维度的数组相乘:A = (10, 1), B = (1, 1, 10, 18) 使用 einsum。

tmp = np.einsum('pi,klpy ->kliy', A, B)

是否需要切换 B 数组的维度才能获得正确的结果?我的意思是:

tmp = np.einsum('pi,lkyp ->kliy', A, B)

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