Первый вектор, N = 3, [1,2, 3]
Вторые N векторов длиной m, m = 2 [[4,5], [6,7], [7,8]]
Итак, N матриц размера (m, m) [[[1,1], [2,2]], [[1,1], [2,2]], [[1,1], [2,2]] ]
Позволяет создавать из них numpy массивов.
import numpy as np
N = 3
m = 2
a = np.array([1,2,3])
b = np.random.randn(N, m)
c = np.random.randn(N, m, m)
Вы видите проблему здесь? Последняя матрица c уже имеет 3 измерения в соответствии с вашими определениями.
Ваш аргумент может быть упрощен.
Допустим, наша окончательная матрица - a = np.zeros((3,2,2))
# 3 измерения, x, y , z 1) Для первого измерения -
a[0,:,:] = 0
# первая ось, первый индекс = 0
a[1,:,:] = 1
# первая ось, 2-й индекс = 1
a[2,:,:] = 2
# первая ось, 3-й индекс = 2
2) Теперь нам нужно заполнить остальные позиции, но размеры не совпадают.
Итак, лучше создать отдельные тензоры для них.