Numpy 数组多维输入顺序

Question

numpy 多维数组的输入顺序是什么？

当我使用 np.ones((3,2,2)) 时，我期望 3 个数组或大小为 2x2，但阅读其他帖子（numpy 文档不清楚）有人声称前两个输入是行和列，其中第三个输入是我想要跨越的维度。

基本上，我想在每个数组前面堆叠多个数组，以创建一个多维张量，类似于火炬与[通道，行，列]的工作方式

谢谢

Answer 1

通常，numpy 将大小为 (A, B, C) 的数组布局为 A 组 BxC 元素，并且每个 BxC 元素都是 B 组 C 元素。因此 x[1,1,1] 在内存中将紧邻 x[1, 1, 2]。

但是你可以说，

np.ones((3, 4, 5), order='F')

，它告诉 numpy 使用 Fortran onrder 而不是 C order。在这种情况下，x[1, 1, 1] 在内存中将与 x[2, 1, 1] 相邻。这不是默认设置。