我想为沿每个轴的每个切片求两个3d数组乘法的总和。
尺寸为(10、180、2)。
我使用了numpy.dot
,numpy.malmul
和numpy.tensordot
,但是没有任何作用
for k in range(0, self.rn):
for n in range(0, self.N):
for m in range(0, self.Vt):
self.Ptot_a[k][n][m] = self.P_c + self.pkm[k][n][m] - 0.1 * sum(
(numpy.tensordot(self.pkm, self.g_km, axes=([i - 1],[i - 1]))) for i in range(0, self.rn))
但是它总是给我一个错误,我最近遇到的错误是:
res = _sum_(a)
ValueError: operands could not be broadcast together with shapes (10,2,10,2) (2,180,2,180)