我有 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/