scala – 存储多维数组/张量的最佳方法

2D random access
  Array of Arrays - 600 M / second
  Multiplication - 1.1 G / second

3D in-order
  Array of Array of Arrays - 2.4G / second
  Multiplication - 2.8 G / second

(etc.)

所以你最好只做数学.

现在的问题是如何进行切片.最初,如果您的尺寸为n1,n2,n3,…以及i1,i2,i3,…的索引,则计算到数组的偏移量

i = i1 + n1*(i2 + n2*(i3 + ... ))

通常选择i1作为最后一个(最里面的)维度(但通常??它应该是最内层循环中最常见的维度).也就是说,如果它是一个(…)数组的数组,你可以将其索引为(…)(i3)(i2)(i1).

现在假设您要切片.首先,您可以为每个索引提供偏移量o1,o2,o3：

i = (i1 + o1) + n1*((i2 + o2) + n2*((i3 + o3) + ...))

然后你会有一个较短的范围(让我们称之为m1,m2,m3,…).

最后,如果你完全消除一个维度 – 比方说,例如,m2 == 1,意味着i2 == 0,你只需简化公式：

i = (i1 + o1 + n1*o2) + (n1+n2)*((i3 + o3) + ... ))

我将把它作为练习留给读者来弄清楚如何一般地做这个,但请注意我们可以存储新的常量o1 n1 * o21和n1 n2所以我们不需要在切片上继续进行数学运算.

最后,如果您允许任意维度,您只需将该数学放入while循环中.不可否认,这确实会让它减慢一点,但你仍然至少可以像使用指针取消引用一样好(几乎在所有情况下).