Jupyter中索引“ pqrs，pt-> sqrt”的张量收缩

Question

我有两个张量分别为（170,170,170,170）和（170,170）的张量I_pqrs和C_st。

我尝试使用np.einsum("pqrt,st->sqrt", I, C)来缩小它们的大小，但是我一直遇到内存错误。我在这里想念东西吗？

代码可以很好地运行到50维。