gpt4 book ai didi

python - 两个二维数组的 np.dot

转载 作者:行者123 更新时间:2023-12-02 02:30:51 26 4
gpt4 key购买 nike

我是 numpy 的新手,很抱歉,如果这听起来很明显,我在发布此内容之前确实尝试通过 stackoverflow 进行搜索..

我有两个长度为 n 的“列表的列表”numpy 数组(在下面的示例中,n = 3)

a = np.array([[1, 2], [3, 4], [5, 6]])
b = np.array([[2, 2], [3, 3], [4, 4]])

我想获得一个一维数组,其中每个相应索引处的列表的点积,即

[(1*2 + 2*2), (3*3 + 4*3), (5*4 + 6*4)]
[6, 21, 44]

我应该怎样做呢?提前致谢!

最佳答案

你可以这样做

np.sum(a*b,axis=1)

关于python - 两个二维数组的 np.dot,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/65059860/

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