gpt4 book ai didi

python - 计算一系列 numpy ndarrays 的外积

转载 作者:行者123 更新时间:2023-11-28 20:35:05 25 4
gpt4 key购买 nike

<分区>

我有一个 3D 点列表 p 存储在一个形状为 (N, 3) 的 ndarray 中。我想计算每个 3d 点的外积:

N = int(1e4)
p = np.random.random((N, 3))
result = np.zeros((N, 3, 3))
for i in range(N):
result[i, :, :] = np.outer(p[i, :], p[i, :])

有没有一种方法可以在没有任何 python 级循环的情况下计算这个外积?问题是 np.outer 不支持任何类似 axis 的参数。

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