gpt4 book ai didi

python - 表达大量点积的更好方法?

转载 作者:太空宇宙 更新时间:2023-11-03 12:21:33 24 4
gpt4 key购买 nike

有没有更好更快的方法来在 numpy 中表达以下点积?我有以下形状:

>>> h.shape
(600L, 400L, 3L)
>>> c.shape
(400L, 3L)

如果可能的话,我想在没有循环的情况下计算以下内容:

ans = np.empty((600, 400))
for i in range(400):
ans[:, i] = h[:, i, :].dot(c[i, :])

我认为通过简单的 reshape 应该是可能的,但我不知道 atm。

最佳答案

你可以使用numpy.einsum

ans = einsum('ijk,jk->ij', h, c)

关于python - 表达大量点积的更好方法?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/16287366/

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