从1D到3D的阵列重塑，元素大小可变

Question

我有一个大小可变的数组。例如，第一个元素的形状为[20,10]，第二个元素的形状为[40,10]，然后第三个元素的形状为[320,10] .....如何在保持该数组的同时将其重塑为3D数组内部数组的大小。