假设test = np.array(5*[np.eye(5), 10*np.eye(5), 15*np.eye(5)])。我在具有shape的数组中有三个矩阵(3, 5, 5)。一般而言，如何重塑test以将三个矩阵合为一体？在该特定示例中，我希望形状为(15, 5)。我想要一种通用的方法，而不必使用一些特别具体的方法np.reshape(test, (15,5))。
您可以在整形中使用-1，以隐式计算所需的尺寸形状：
test = test.reshape(-1, test.shape[-1])
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句