Given a 3D matrix of shape[3, 5, 7] that is implemented using a 1D array, I can calculate the index of any element (e.g. [1, 0, 6]) as follows:
index = 1 * 5 * 7 + 0 * 7 + 6
In general, if I have a K x L x M matrix, I can access index[i][j][k] by calculating iLM + jM + k to yield the index. Likewise, when working with a 4D matrix[h][i][j][k], I can achieve the same by calculating hLMN + iMN + j*N + k.
If I have both the index ([i, j, k]) and the shape ([K, L, M]) stored in two separate arrays of equal length, how could I dynamically generate the above calculations for any two such arrays of arbitrary length 3 or greater?
Been racking my brain on this for hours. Please point me in the right direction.