python - 如何展平 numpy 数组的最后两个维度？

Question

例如，给定一个尺寸为 (132, 82, 100) 的 numpy 数组，结果尺寸将为 (132, 8200)

标签： pythonnumpy

Answer 1

您可以使用reshape()函数来更改数组的形状。使用 -1 作为 reshape 的参数告诉 numpy 推断那里的尺寸。

arr = np.zeros((132, 82, 100))
arr = arr.reshape(*arr.shape[:-2], -1)
print(arr.shape)
# Out: (132, 8200)