У меня есть два тензора I_pqrs и C_st с размерами (170,170,170,170) и (170,170) соответственно.
Я пытался свернуть их, используя np.einsum("pqrt,st->sqrt", I, C), но постоянно получаю ошибку памяти. Я что-то здесь упускаю?
Код работает нормально до размеров 50.