gpt4 book ai didi

python - 两个 numpy 数组的逐元素减法

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

我有两个 numpy 数组 'a' 和 'b' 分别具有 (327600,5) 和 (3,5) 的维度。我想对两个数组进行逐元素减法。如果没有循环,我将如何去做呢?

当我执行 a[0] - b 时,我得到一个 (3,5) 数组。我想得到一个 (327600,3,5) 数组作为减法后的结果。

最佳答案

a[:, np.newaxis] - b

您需要查看 a 在形状中有一个额外的 length-1 轴,以便这些轴对齐以进行广播。

关于python - 两个 numpy 数组的逐元素减法,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/33793163/

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