你如何展平 numpy 数组的最后两个维度?

乔希

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

舒巴姆·洛希亚

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

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

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章