gpt4 book ai didi

python - Numpy:确定广播后的形状

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

我有 2 个 numpy 数组的形状。我想检查是否可以添加这些数组并确定结果形状。

有什么方法可以快速做到这一点并覆盖所有边缘情况吗?

最佳答案

如果您有实际的 NumPy 对象,您可以使用 NumPy 的逻辑来确定这一点(添加需要广播,请参阅 here ):

def checkShapes(a, b):
try:
return (np.broadcast(a, b)).shape
except:
return (-1,)

关于python - Numpy:确定广播后的形状,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/36949908/

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