主页 / 问题 / 1196429
Accepted
Wrong Wizzli
Asked:2021-10-28 02:58:35 +0000 UTC2021-10-28 02:58:35 +0000 UTC

如何转换数组一部分的维度？

  • 772
A = np.ones((14, 32, 3, 3))

假设我们有一个 4 维 NumPy 数组。

np.ravel是否可以沿数组的最后三个轴应用该函数或类似函数А？

也就是说，结果应该是一个维度为 的矩阵(14, 32 * 3 * 3))。

python

1 个回答

  1. Best Answer

    您可以使用以下方法ndarray.reshape()：

    A = np.ones((14, 32, 3, 3))
res = A.reshape(-1, np.prod(A.shape[1:]))

    结果：

    res.shape
>>> (14, 288)
    • 4

相关问题