我有四个张量

> H(h,r)
> A(a,r)
> D(d,r)
> T(a,t,r)

对于a中的每个i,存在形状(t,r)的对应T [i].

我需要做一个np.einsum来产生以下结果(pred)：

pred = np.einsum('hr,ar,dr,tr ->hadt',H,A,D,T[0])
for i in range(a):
    pred[:,i:i+1,:,:] = np.einsum('hr,tr ->HADT',A[i:i+1],T[i])

但是,我想在不使用for循环的情况下进行此计算.原因是我正在使用autograd,它目前不适用于项目分配！