如何组合Numpy Array的最后2个维度

Question

我已经看过How to combine dimensions in numpy array?了

并尝试过，但当我尝试

imgs.reshape(img.shape[:-2]+(-1))

我假设是逻辑扩展，我得到一个错误：

can only concatenate tuple (not "int") to tuple

我将对来自链接问题的已接受答案进行评论，并直接向他们寻求帮助，而不是为此创建新的帖子，但我没有必要的评论声誉。

编辑：例如，我有一个形状为(x,y,8,8)的输入数组，其中x和y可以根据输入函数的图像而改变。我想把它变成一个形状(x,y,64)阵列。在这种情况下，64永远不会改变。

Answer 1

您必须添加逗号才能将其识别为元组。尝试

imgs.reshape(img.shape[:-2] + (-1,))

或者，使用拆包：

imgs.reshape((*img.shape[:-2], -1))