У меня есть два 2D JaggedArrays одинаковой длины (по оси = 0), которые я хотел бы сложить по оси = 1. Например:
a = awkward.fromiter([[0], [3, 4]])
b = awkward.fromiter([[1, 2], [5]])
и я хочу получить этот JaggedArray:
[ [ [0], [1, 2] ],
[ [3, 4], [5] ] ]
Я могу получить именно это с помощью следующего:
awkward.fromiter(np.stack([a, b], axis=1))
но есть способ выполнить sh прямо в awkward, не используя fromiter() для вложенных numpy массивов?