谁能告诉我如何在python中做张量积。我的意思是,例如,我有5 (6,1) (类: numpy.ndarray)数组,我需要这5 (6,1)数组的张量积,所以我的输出将是(6,6,6,6,6)张量。
发布于 2020-09-21 14:14:25
下面的方法对你有效吗?
import numpy as np
a1 = np.arange(6)
a2 = np.arange(6)
a3 = np.arange(6)
a4 = np.arange(6)
a5 = np.arange(6)
oo = np.einsum('i,j,k,l,m',a1,a2,a3,a4,a5)https://stackoverflow.com/questions/63986495
复制相似问题